From 9780aefa9c78ae97a897661968bb6a49a80a8180 Mon Sep 17 00:00:00 2001 From: "Anders H. Rebner" <anderhre@stud.ntnu.no> Date: Mon, 16 Mar 2020 16:41:47 +0100 Subject: [PATCH] #58 test_database.py modifisert etter ny tabellstruktur --- test/test_database.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/test_database.py b/test/test_database.py index e43cdc2..d1c1897 100644 --- a/test/test_database.py +++ b/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 -- GitLab