diff options
Diffstat (limited to 'backend/tests/test_gen_disp_data.py')
| -rw-r--r-- | backend/tests/test_gen_disp_data.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/backend/tests/test_gen_disp_data.py b/backend/tests/test_gen_disp_data.py index 0d54eb0..8fe13e4 100644 --- a/backend/tests/test_gen_disp_data.py +++ b/backend/tests/test_gen_disp_data.py @@ -1,5 +1,6 @@ import unittest -import tempfile, os +import tempfile +import os from tests.common import createTestDbTable, readTestDbTable from hist_data.gen_disp_data import genData @@ -58,9 +59,11 @@ class TestGenData(unittest.TestCase): (7, 70), } ) + # Run 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'), |