You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
867 B
29 lines
867 B
6 years ago
|
import unittest
|
||
|
|
import pandas as pd
|
||
|
|
from analytic_unit_manager import prepare_data
|
||
|
|
import models
|
||
|
|
|
||
|
|
class TestDataset(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_models_with_corrupted_dataframe(self):
|
||
|
|
data = [[1523889000000 + i, float('nan')] for i in range(10)]
|
||
|
|
dataframe = pd.DataFrame(data, columns=['timestamp', 'value'])
|
||
|
|
segments = []
|
||
|
|
|
||
|
|
model_instances = [
|
||
|
|
models.JumpModel(),
|
||
|
|
models.DropModel(),
|
||
|
|
models.GeneralModel(),
|
||
|
|
models.PeakModel(),
|
||
|
|
models.TroughModel()
|
||
|
|
]
|
||
|
|
try:
|
||
|
|
for model in model_instances:
|
||
|
|
model_name = model.__class__.__name__
|
||
|
|
model.fit(dataframe, segments, dict())
|
||
|
|
except ValueError:
|
||
|
|
self.fail('Model {} raised unexpectedly'.format(model_name))
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|