diff --git a/tests/test_account.py b/tests/test_account.py index 343631f..578bc48 100644 --- a/tests/test_account.py +++ b/tests/test_account.py @@ -25,8 +25,8 @@ from click.testing import Result from flask import Flask from flask.testing import FlaskCliRunner -from testlib import get_client from test_site import create_app +from testlib import get_client class AccountCommandTestCase(unittest.TestCase): diff --git a/tests/test_base_account.py b/tests/test_base_account.py index 800c57c..67e900b 100644 --- a/tests/test_base_account.py +++ b/tests/test_base_account.py @@ -24,8 +24,8 @@ from click.testing import Result from flask import Flask from flask.testing import FlaskCliRunner -from testlib import get_client from test_site import create_app +from testlib import get_client class BaseAccountCommandTestCase(unittest.TestCase):