diff options
| author | Terry Truong <terry06890@gmail.com> | 2023-01-14 22:03:57 +1100 |
|---|---|---|
| committer | Terry Truong <terry06890@gmail.com> | 2023-01-14 22:03:57 +1100 |
| commit | ebe74b640adaed5382046a49b1c2d6a48b31ebfd (patch) | |
| tree | 8ca2b824d3667d9a6e5ef4e8b2d6b739f6692aaf /backend/tests/test_gen_disp_data.py | |
| parent | a8ea534f13cc23dfab25c6d856f9d9833dd000ba (diff) | |
Add img_dist and img_disp db tables
Having separate event counts for events with images resolves some
redundant server querying, and makes event count indicators moree
representative.
Diffstat (limited to 'backend/tests/test_gen_disp_data.py')
| -rw-r--r-- | backend/tests/test_gen_disp_data.py | 49 |
1 files changed, 48 insertions, 1 deletions
diff --git a/backend/tests/test_gen_disp_data.py b/backend/tests/test_gen_disp_data.py index 792a9b2..0d54eb0 100644 --- a/backend/tests/test_gen_disp_data.py +++ b/backend/tests/test_gen_disp_data.py @@ -46,8 +46,21 @@ class TestGenData(unittest.TestCase): (11, 45), } ) + createTestDbTable( + dbFile, + 'CREATE TABLE event_imgs (id INT PRIMARY KEY, img_id INT)', + 'INSERT INTO event_imgs VALUES (?, ?)', + { + (1, 10), + (4, 40), + (5, 50), + (6, 60), + (7, 70), + } + ) # Run - genData(dbFile, [10, 1, MONTH_SCALE, DAY_SCALE], 2) + genData(dbFile, [10, 1, MONTH_SCALE, DAY_SCALE], 2, False) + genData(dbFile, [10, 1, MONTH_SCALE, DAY_SCALE], 2, True) # Check self.assertEqual( readTestDbTable(dbFile, 'SELECT * FROM events'), @@ -117,3 +130,37 @@ class TestGenData(unittest.TestCase): (2, DAY_SCALE, 2452607), } ) + self.assertEqual( + readTestDbTable(dbFile, 'SELECT scale, unit, count FROM img_dist'), + { + (10, 190, 5), + (1, 1900, 4), + (1, 1901, 1), + (MONTH_SCALE, gregorianToJdn(1900, 1, 1), 3), + (MONTH_SCALE, gregorianToJdn(1901, 1, 1), 1), + (MONTH_SCALE, julianToJdn(1900, 10, 1), 1), + (DAY_SCALE, gregorianToJdn(1900, 1, 1), 2), + (DAY_SCALE, gregorianToJdn(1900, 1, 10), 1), + (DAY_SCALE, julianToJdn(1900, 10, 1), 1), + (DAY_SCALE, gregorianToJdn(1901, 1, 1), 1), + } + ) + self.assertEqual( + readTestDbTable(dbFile, 'SELECT id, scale, unit FROM img_disp'), + { + (5, 10, 190), + (7, 10, 190), + (5, 1, 1900), + (7, 1, 1900), + (4, 1, 1901), + (1, MONTH_SCALE, 2415021), + (7, MONTH_SCALE, 2415021), + (4, MONTH_SCALE, 2415386), + (5, MONTH_SCALE, 2415307), + (1, DAY_SCALE, 2415021), + (7, DAY_SCALE, 2415021), + (6, DAY_SCALE, 2415030), + (4, DAY_SCALE, 2415386), + (5, DAY_SCALE, 2415307), + } + ) |