➕Added unittest for pipeline algo
This commit is contained in:
parent
a2b59b5bf4
commit
2972674903
|
|
@ -0,0 +1,40 @@
|
|||
import json
|
||||
import unittest
|
||||
from app import aggregate_salaries, MongoDB, Settings
|
||||
|
||||
settings = Settings()
|
||||
|
||||
with open(r'tests\samples.json', 'r') as f:
|
||||
data = json.load(f)
|
||||
|
||||
class TestAggregateSalaries(unittest.IsolatedAsyncioTestCase):
|
||||
async def async_test(self, json_in, json_out, collection):
|
||||
result = await aggregate_salaries(collection, **json_in)
|
||||
self.assertEqual(json_out, result)
|
||||
|
||||
async def test_month(self):
|
||||
client = MongoDB(str(settings.DB_URI))
|
||||
db = client.client[settings.DATABASE_NAME]
|
||||
collection = db[settings.COLLECTION_NAME]
|
||||
json_in = data['month']['in']
|
||||
json_out = data['month']['out']
|
||||
await self.async_test(json_in, json_out, collection)
|
||||
|
||||
async def test_day(self):
|
||||
client = MongoDB(str(settings.DB_URI))
|
||||
db = client.client[settings.DATABASE_NAME]
|
||||
collection = db[settings.COLLECTION_NAME]
|
||||
json_in = data['day']['in']
|
||||
json_out = data['day']['out']
|
||||
await self.async_test(json_in, json_out, collection)
|
||||
|
||||
async def test_hour(self):
|
||||
client = MongoDB(str(settings.DB_URI))
|
||||
db = client.client[settings.DATABASE_NAME]
|
||||
collection = db[settings.COLLECTION_NAME]
|
||||
json_in = data['hour']['in']
|
||||
json_out = data['hour']['out']
|
||||
await self.async_test(json_in, json_out, collection)
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
|
@ -0,0 +1,60 @@
|
|||
{
|
||||
"month": {
|
||||
"in":{
|
||||
"dt_from": "2022-09-01T00:00:00",
|
||||
"dt_upto": "2022-12-31T23:59:00",
|
||||
"group_type": "month"
|
||||
},
|
||||
"out": {
|
||||
"dataset": [5906586, 5515874, 5889803, 6092634],
|
||||
"labels": ["2022-09-01T00:00:00", "2022-10-01T00:00:00", "2022-11-01T00:00:00", "2022-12-01T00:00:00"]
|
||||
}
|
||||
},
|
||||
"day": {
|
||||
"in":{
|
||||
"dt_from": "2022-10-01T00:00:00",
|
||||
"dt_upto": "2022-11-30T23:59:00",
|
||||
"group_type": "day"
|
||||
},
|
||||
"out": {
|
||||
"dataset": [0, 0, 0, 195028, 190610, 193448, 203057, 208605, 191361, 186224, 181561, 195264, 213854, 194070,208372, 184966, 196745, 185221, 196197, 200647, 196755, 221695, 189114, 204853, 194652, 188096, 215141,
|
||||
185000, 206936, 200164, 188238, 195279, 191601, 201722, 207361, 184391, 203336, 205045, 202717, 182251,
|
||||
185631, 186703, 193604, 204879, 201341, 202654, 183856, 207001, 204274, 204119, 188486, 191392, 184199,
|
||||
202045, 193454, 198738, 205226, 188764, 191233, 193167, 205334],
|
||||
"labels": ["2022-10-01T00:00:00", "2022-10-02T00:00:00", "2022-10-03T00:00:00", "2022-10-04T00:00:00",
|
||||
"2022-10-05T00:00:00", "2022-10-06T00:00:00", "2022-10-07T00:00:00", "2022-10-08T00:00:00",
|
||||
"2022-10-09T00:00:00", "2022-10-10T00:00:00", "2022-10-11T00:00:00", "2022-10-12T00:00:00",
|
||||
"2022-10-13T00:00:00", "2022-10-14T00:00:00", "2022-10-15T00:00:00", "2022-10-16T00:00:00",
|
||||
"2022-10-17T00:00:00", "2022-10-18T00:00:00", "2022-10-19T00:00:00", "2022-10-20T00:00:00",
|
||||
"2022-10-21T00:00:00", "2022-10-22T00:00:00", "2022-10-23T00:00:00", "2022-10-24T00:00:00",
|
||||
"2022-10-25T00:00:00", "2022-10-26T00:00:00", "2022-10-27T00:00:00", "2022-10-28T00:00:00",
|
||||
"2022-10-29T00:00:00", "2022-10-30T00:00:00", "2022-10-31T00:00:00", "2022-11-01T00:00:00",
|
||||
"2022-11-02T00:00:00", "2022-11-03T00:00:00", "2022-11-04T00:00:00", "2022-11-05T00:00:00",
|
||||
"2022-11-06T00:00:00", "2022-11-07T00:00:00", "2022-11-08T00:00:00", "2022-11-09T00:00:00",
|
||||
"2022-11-10T00:00:00", "2022-11-11T00:00:00", "2022-11-12T00:00:00", "2022-11-13T00:00:00",
|
||||
"2022-11-14T00:00:00", "2022-11-15T00:00:00", "2022-11-16T00:00:00", "2022-11-17T00:00:00",
|
||||
"2022-11-18T00:00:00", "2022-11-19T00:00:00", "2022-11-20T00:00:00", "2022-11-21T00:00:00",
|
||||
"2022-11-22T00:00:00", "2022-11-23T00:00:00", "2022-11-24T00:00:00", "2022-11-25T00:00:00",
|
||||
"2022-11-26T00:00:00", "2022-11-27T00:00:00", "2022-11-28T00:00:00", "2022-11-29T00:00:00",
|
||||
"2022-11-30T00:00:00"]
|
||||
}
|
||||
},
|
||||
"hour": {
|
||||
"in":{
|
||||
"dt_from": "2022-02-01T00:00:00",
|
||||
"dt_upto": "2022-02-02T00:00:00",
|
||||
"group_type": "hour"
|
||||
},
|
||||
"out": {
|
||||
"dataset": [8177, 8407, 4868, 7706, 8353, 7143, 6062, 11800, 4077, 8820, 4788, 11045, 13048, 2729, 4038, 9888,
|
||||
7490, 11644, 11232, 12177, 2741, 5341, 8730, 4718, 0],
|
||||
"labels": ["2022-02-01T00:00:00", "2022-02-01T01:00:00", "2022-02-01T02:00:00", "2022-02-01T03:00:00",
|
||||
"2022-02-01T04:00:00", "2022-02-01T05:00:00", "2022-02-01T06:00:00", "2022-02-01T07:00:00",
|
||||
"2022-02-01T08:00:00", "2022-02-01T09:00:00", "2022-02-01T10:00:00", "2022-02-01T11:00:00",
|
||||
"2022-02-01T12:00:00", "2022-02-01T13:00:00", "2022-02-01T14:00:00", "2022-02-01T15:00:00",
|
||||
"2022-02-01T16:00:00", "2022-02-01T17:00:00", "2022-02-01T18:00:00", "2022-02-01T19:00:00",
|
||||
"2022-02-01T20:00:00", "2022-02-01T21:00:00", "2022-02-01T22:00:00", "2022-02-01T23:00:00",
|
||||
"2022-02-02T00:00:00"]
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue