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