diff options
Diffstat (limited to 'backend/tests/enwiki/test_gen_img_data.py')
| -rw-r--r-- | backend/tests/enwiki/test_gen_img_data.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/backend/tests/enwiki/test_gen_img_data.py b/backend/tests/enwiki/test_gen_img_data.py index d18dddf..91ba481 100644 --- a/backend/tests/enwiki/test_gen_img_data.py +++ b/backend/tests/enwiki/test_gen_img_data.py @@ -1,5 +1,6 @@ import unittest -import tempfile, os +import tempfile +import os from tests.common import createTestDbTable, readTestDbTable from hist_data.enwiki.gen_img_data import getInputPageIdsFromDb, genData @@ -24,6 +25,7 @@ class TestGetInputPageIdsFromDb(unittest.TestCase): (5, 'Marie Curie', 2403277, None, 2427622, None, 1, 'human'), } ) + # Create temp dump-index db indexDb = os.path.join(tempDir, 'dump_index.db') createTestDbTable( @@ -38,6 +40,7 @@ class TestGetInputPageIdsFromDb(unittest.TestCase): ('Autism',25,0,-1), } ) + # Run pageIds = getInputPageIdsFromDb(dbFile, indexDb) # Check @@ -58,6 +61,7 @@ class TestGenData(unittest.TestCase): ('Autism',25,0,-1), } ) + # Run imgDb = os.path.join(tempDir, 'imgData.db') genData({10, 25}, TEST_DUMP_FILE, indexDb, imgDb) @@ -69,6 +73,7 @@ class TestGenData(unittest.TestCase): (25, 'Autism', 'Autism-stacking-cans 2nd edit.jpg'), } ) + # Run with updated page-ids set genData({13, 10}, TEST_DUMP_FILE, indexDb, imgDb) # Check |