From 149dc178c491d8e447a05ff3705fdc6ceddf129e Mon Sep 17 00:00:00 2001 From: Terry Truong Date: Sun, 2 Oct 2022 14:43:02 +1100 Subject: Add title column to img_data.db --- backend/tests/enwiki/test_gen_img_data.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'backend/tests/enwiki/test_gen_img_data.py') 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'), } ) -- cgit v1.2.3