aboutsummaryrefslogtreecommitdiff
path: root/backend/tests/test_gen_desc_data.py
diff options
context:
space:
mode:
authorTerry Truong <terry06890@gmail.com>2023-01-29 11:30:47 +1100
committerTerry Truong <terry06890@gmail.com>2023-01-29 11:30:47 +1100
commit8781fdb2b8c530a6c1531ae9e82221eb062e34fb (patch)
treeffd824aa9b945d69b47f012617ee13d98764d078 /backend/tests/test_gen_desc_data.py
parentf5e87ae628bab0eef97b3e3e62f6d71cca9c99c0 (diff)
Adjust backend coding style
Add line spacing, section comments, and import consistency
Diffstat (limited to 'backend/tests/test_gen_desc_data.py')
-rw-r--r--backend/tests/test_gen_desc_data.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/backend/tests/test_gen_desc_data.py b/backend/tests/test_gen_desc_data.py
index cc0582d..8d21978 100644
--- a/backend/tests/test_gen_desc_data.py
+++ b/backend/tests/test_gen_desc_data.py
@@ -1,5 +1,6 @@
import unittest
-import tempfile, os
+import tempfile
+import os
from tests.common import createTestDbTable, readTestDbTable
from tol_data.gen_desc_data import genData
@@ -37,6 +38,7 @@ class TestGenData(unittest.TestCase):
('<http://dbpedia.org/resource/Three>', 'Three from dbp'),
}
)
+
# Create temp enwiki db
enwikiDb = os.path.join(tempDir, 'enwiki_descs.db')
createTestDbTable(
@@ -70,6 +72,7 @@ class TestGenData(unittest.TestCase):
(5, 'Five from enwiki'),
}
)
+
# Create temp tree-of-life db
dbFile = os.path.join(tempDir, 'data.db')
createTestDbTable(
@@ -86,8 +89,10 @@ class TestGenData(unittest.TestCase):
('seventh', 7),
}
)
+
# Run
genData(dbpediaDb, enwikiDb, dbFile)
+
# Check
self.assertEqual(
readTestDbTable(dbFile, 'SELECT wiki_id, desc, from_dbp from descs'),