diff options
Diffstat (limited to 'tests/test_db.py')
| -rw-r--r-- | tests/test_db.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_db.py b/tests/test_db.py index 6ac4635..4e2f103 100644 --- a/tests/test_db.py +++ b/tests/test_db.py @@ -1,7 +1,7 @@ import sqlite3 import pytest -from flaskr.db import get_db +from ustayml.db import get_db def test_get_close_db(app): @@ -27,7 +27,7 @@ def test_init_db_command(runner, monkeypatch): def fake_init_db(): Recorder.called = True - monkeypatch.setattr('flaskr.db.init_db', fake_init_db) + monkeypatch.setattr('ustayml.db.init_db', fake_init_db) result = runner.invoke(args=['init-db']) assert 'Initialized' in result.output assert Recorder.called
\ No newline at end of file |