From 9f9442f2488c89d47f9f0191fcd9ead7a8f8c571 Mon Sep 17 00:00:00 2001
From: morkolai <nikolai-mork@live.no>
Date: Thu, 27 Feb 2020 10:43:48 +0100
Subject: [PATCH] =?UTF-8?q?#9=20Fullf=C3=B8rt=20test=20for=20CodeBook?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
test/test_database.py | 19 ++++++++++++++-----
1 file changed, 14 insertions(+), 5 deletions(-)
diff --git a/test/test_database.py b/test/test_database.py
index bf1d8a8..37239aa 100644
--- a/test/test_database.py
+++ b/test/test_database.py
@@ -105,16 +105,25 @@ class DatabaseTest(unittest.TestCase):
def test_get_codebook(self):
"""Assert function get_codebook works as expected."""
-
+ # Get test-data from json
file_path = os.path.join(TESTDATA_PATH, "codebook.json")
with open(file_path, "r") as file:
- data = json.load(file)
+ expected = json.load(file)
file.close()
- stmt = 'SELECT ALL FROM CodeBook'
- queried = self.database.conn.execute(stmt).fetchall()
+ # Get data from db
+ stmt = 'SELECT * FROM CodeBook'
+ actual = self.database.conn.execute(stmt).fetchall()
+
+ # Check same lenght
+ self.assertEqual(len(expected), len(actual))
-
+ # Check total equality
+ for i in range(len(expected)):
+ self.assertEqual(expected[i]['word'], actual[i][0])
+ self.assertEqual(expected[i]['category'], actual[i][1])
+ self.assertEqual(expected[i]['type'], actual[i][2])
+ self.assertEqual(expected[i]['code'], actual[i][3])
if __name__ == '__main__':
--
GitLab