diff --git a/soitool/database.py b/soitool/database.py index 4241f2d8e7533c2672068672dc537846227cf1fa..0e00dd73359f44285e2146a3202eae06f7d54d67 100644 --- a/soitool/database.py +++ b/soitool/database.py @@ -44,9 +44,9 @@ class Database: Database can be run in testmode, this will create a seperate database-file. """ - def __init__(self, test_mode=False): + def __init__(self, testmode=False): - if test_mode: + if testmode: db_path = TESTDBPATH else: db_path = DBPATH diff --git a/test/test_database.py b/test/test_database.py index e413346cafa2a121314d034fda325f781de82827..831919169c50f1a39919cc8225cda35f9ee39db2 100644 --- a/test/test_database.py +++ b/test/test_database.py @@ -16,7 +16,7 @@ class DatabaseTest(unittest.TestCase): def setUp(self): """Connect to/create database.""" - self.database = Database(test_mode=True) + self.database = Database(testmode=True) self.addCleanup(self.delete_db) def test_connection(self):