mirror of
https://github.com/zulip/zulip.git
synced 2025-10-24 16:43:57 +00:00
6e425814bf
The code in fixtures.py is only called from populate_analytics_db, and is only used for generating pretty fixture data for manual testing. This commit adds tests for a few things that were easy to add tests for, and provides some minimal coverage of the file, but is not meant to be comprehensive.
32 lines
1.5 KiB
Python
32 lines
1.5 KiB
Python
from zerver.lib.test_classes import ZulipTestCase
|
|
|
|
from analytics.lib.counts import CountStat
|
|
from analytics.lib.fixtures import generate_time_series_data
|
|
|
|
# A very light test suite; the code being tested is not run in production.
|
|
class TestFixtures(ZulipTestCase):
|
|
def test_deterministic_settings(self):
|
|
# type: () -> None
|
|
# test basic business_hour / non_business_hour calculation
|
|
# test we get an array of the right length with frequency=CountStat.DAY
|
|
data = generate_time_series_data(
|
|
days=7, business_hours_base=20, non_business_hours_base=15, spikiness=0)
|
|
self.assertEqual(data, [400, 400, 400, 400, 400, 360, 360])
|
|
|
|
data = generate_time_series_data(
|
|
days=1, business_hours_base=2000, non_business_hours_base=1500,
|
|
growth=2, spikiness=0, frequency=CountStat.HOUR)
|
|
# test we get an array of the right length with frequency=CountStat.HOUR
|
|
self.assertEqual(len(data), 24)
|
|
# test that growth doesn't affect the first data point
|
|
self.assertEqual(data[0], 2000)
|
|
# test that the last data point is growth times what it otherwise would be
|
|
self.assertEqual(data[-1], 1500*2)
|
|
|
|
# test autocorrelation == 1, since that's the easiest value to test
|
|
data = generate_time_series_data(
|
|
days=1, business_hours_base=2000, non_business_hours_base=2000,
|
|
autocorrelation=1, frequency=CountStat.HOUR)
|
|
self.assertEqual(data[0], data[1])
|
|
self.assertEqual(data[0], data[-1])
|