#58 test_database.py modifisert etter ny tabellstruktur

test/test_database.py

@@ -90,7 +90,7 @@ class DatabaseTest(unittest.TestCase):
         f.close()
 
         # Get data from db
-        stmt = "SELECT * FROM CodeBook"
+        stmt = "SELECT * FROM CodeBook ORDER BY Word"
         actual = self.database.conn.execute(stmt).fetchall()
 
         # Check same lenght
@@ -159,10 +159,10 @@ class DatabaseTest(unittest.TestCase):
         # Fill expected with only small codebook entries
         expected = []
         for entry in data:
-            if entry["type"] == 1:
+            if entry["type"] == 'Liten':
                 expected.append(entry)
         # Get small codebook from db
-        actual = self.database.get_codebook(codebook_size=CodebookSize.FULL)
+        actual = self.database.get_codebook(codebook_size=CodebookSize.SMALL)
         # Compare lenght
         self.assertEqual(len(expected), len(actual))
         # Get expected length of codes