diff options
Diffstat (limited to 'backend/tests/enwiki/test_gen_img_data.py')
| -rw-r--r-- | backend/tests/enwiki/test_gen_img_data.py | 31 |
1 files changed, 9 insertions, 22 deletions
diff --git a/backend/tests/enwiki/test_gen_img_data.py b/backend/tests/enwiki/test_gen_img_data.py index 93bb196..d18dddf 100644 --- a/backend/tests/enwiki/test_gen_img_data.py +++ b/backend/tests/enwiki/test_gen_img_data.py @@ -9,7 +9,7 @@ TEST_DUMP_FILE = os.path.join(os.path.dirname(__file__), 'sample_enwiki_pages_ar class TestGetInputPageIdsFromDb(unittest.TestCase): def test_get(self): with tempfile.TemporaryDirectory() as tempDir: - # Create temp tree-of-life db + # Create temp history db dbFile = os.path.join(tempDir, 'data.db') createTestDbTable( dbFile, @@ -17,24 +17,11 @@ class TestGetInputPageIdsFromDb(unittest.TestCase): 'start INT, start_upper INT, end INT, end_upper INT, fmt INT, ctg TEXT)', 'INSERT INTO events VALUES (?, ?, ?, ?, ?, ?, ?, ?)', { - (1, 'Belgium', 2389729, None, None, None, 2, 'country'), - (2, 'George Washington', 2353711, None, 2378478, None, 2, 'human'), - (3, 'Douglas Adams', 2434082, None, 2452040, None, 2, 'human'), - (4, 'World War II', 2429507, None, 2431700, None, 2, 'event'), - (5, 'Marie Curie', 2403277, None, 2427622, None, 2, 'human'), - } - ) - # Create temp pageviews db - pageviewDb = os.path.join(tempDir, 'pageview_data.db') - createTestDbTable( - pageviewDb, - 'CREATE TABLE views (title TEXT PRIMARY KEY, id INT, views INT)', - 'INSERT INTO views VALUES (?, ?, ?)', - { - ('George Washington', 2, 8), - ('Marie Curie', 5, 10), - ('Douglas Adams', 3, 5), - ('Belgium', 1, 100), + (1, 'Belgium', 2389729, None, None, None, 1, 'country'), + (2, 'George Washington', 2353711, None, 2378478, None, 1, 'human'), + (3, 'Douglas Adams', 2434082, None, 2452040, None, 1, 'human'), + (4, 'World War II', 2429507, None, 2431700, None, 1, 'event'), + (5, 'Marie Curie', 2403277, None, 2427622, None, 1, 'human'), } ) # Create temp dump-index db @@ -46,15 +33,15 @@ class TestGetInputPageIdsFromDb(unittest.TestCase): { ('Belgium',10,0,-1), ('George Washington',20,0,-1), - ('Douglas Adamns',30,0,-1), + ('Douglas Adams',30,0,-1), ('Marie Curie',50,0,-1), ('Autism',25,0,-1), } ) # Run - pageIds = getInputPageIdsFromDb(dbFile, pageviewDb, indexDb, 2) + pageIds = getInputPageIdsFromDb(dbFile, indexDb) # Check - self.assertEqual(pageIds, {50, 20, 10}) + self.assertEqual(pageIds, {10, 20, 30, 50}) class TestGenData(unittest.TestCase): def test_gen(self): |