diff options
| author | Terry Truong <terry06890@gmail.com> | 2022-10-02 14:43:02 +1100 |
|---|---|---|
| committer | Terry Truong <terry06890@gmail.com> | 2022-10-02 21:13:10 +1100 |
| commit | 149dc178c491d8e447a05ff3705fdc6ceddf129e (patch) | |
| tree | 7194380d61c96e77366936a68c14026ab17ac04e /backend/tests | |
| parent | d70b96295d768aa0c80bf66639ad7a56bdef92a8 (diff) | |
Add title column to img_data.db
Diffstat (limited to 'backend/tests')
| -rw-r--r-- | backend/tests/enwiki/test_gen_img_data.py | 14 | ||||
| -rw-r--r-- | backend/tests/wikidata/__init__.py | 0 |
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 |