aboutsummaryrefslogtreecommitdiff
path: root/backend/tests
diff options
context:
space:
mode:
Diffstat (limited to 'backend/tests')
-rw-r--r--backend/tests/enwiki/test_gen_img_data.py14
-rw-r--r--backend/tests/wikidata/__init__.py0
2 files changed, 7 insertions, 7 deletions
diff --git a/backend/tests/enwiki/test_gen_img_data.py b/backend/tests/enwiki/test_gen_img_data.py
index dc77fe6..93bb196 100644
--- a/backend/tests/enwiki/test_gen_img_data.py
+++ b/backend/tests/enwiki/test_gen_img_data.py
@@ -76,20 +76,20 @@ class TestGenData(unittest.TestCase):
genData({10, 25}, TEST_DUMP_FILE, indexDb, imgDb)
# Check
self.assertEqual(
- readTestDbTable(imgDb, 'SELECT page_id, img_name from page_imgs'),
+ readTestDbTable(imgDb, 'SELECT page_id, title, img_name from page_imgs'),
{
- (10, None),
- (25, 'Autism-stacking-cans 2nd edit.jpg'),
+ (10, None, None),
+ (25, 'Autism', 'Autism-stacking-cans 2nd edit.jpg'),
}
)
# Run with updated page-ids set
genData({13, 10}, TEST_DUMP_FILE, indexDb, imgDb)
# Check
self.assertEqual(
- readTestDbTable(imgDb, 'SELECT page_id, img_name from page_imgs'),
+ readTestDbTable(imgDb, 'SELECT page_id, title, img_name from page_imgs'),
{
- (10, None),
- (13, None),
- (25, 'Autism-stacking-cans 2nd edit.jpg'),
+ (10, None, None),
+ (13, None, None),
+ (25, 'Autism', 'Autism-stacking-cans 2nd edit.jpg'),
}
)
diff --git a/backend/tests/wikidata/__init__.py b/backend/tests/wikidata/__init__.py
deleted file mode 100644
index e69de29..0000000
--- a/backend/tests/wikidata/__init__.py
+++ /dev/null