From b6c3a8e6a808e455b711cdf7276ecc911d4f0693 Mon Sep 17 00:00:00 2001
From: Alexander Velikiy <39257464+VargBurz@users.noreply.github.com>
Date: Wed, 12 Feb 2020 12:51:08 +0300
Subject: [PATCH] Test for Anomaly Detector (#839)

---
 analytics/tests/test_detectors.py | 46 +++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/analytics/tests/test_detectors.py b/analytics/tests/test_detectors.py
index ab4115b..573cc9c 100644
--- a/analytics/tests/test_detectors.py
+++ b/analytics/tests/test_detectors.py
@@ -211,5 +211,51 @@ class TestAnomalyDetector(unittest.TestCase):
         result = [{ 'from': 1523889000010, 'to': 1523889000010 }]
         self.assertEqual(result, detected_segments)
 
+    def test_get_bounds_for_segment(self):
+        detector = anomaly_detector.AnomalyDetector('test_id')
+        peak_segment = pd.Series([1,2,3,4,3,2,1])
+        trough_segment = pd.Series([4,3,2,1,2,3,4])
+        expected_peak_segment_results = {
+            'max_value': 3,
+            'min_value': 1.5
+        }
+        expected_trough_segment_results = {
+            'max_value': 3.5,
+            'min_value': 2.75
+        }
+        peak_detector_result = detector.get_bounds_for_segment(peak_segment)
+        trough_detector_result = detector.get_bounds_for_segment(trough_segment)
+
+        self.assertGreaterEqual(
+            max(peak_detector_result[0]),
+            expected_peak_segment_results['max_value']
+        )
+        self.assertLessEqual(
+            max(peak_detector_result[1]),
+            expected_peak_segment_results['min_value']
+        )
+        self.assertGreaterEqual(
+            max(trough_detector_result[0]),
+            expected_trough_segment_results['max_value']
+        )
+        self.assertLessEqual(
+            max(trough_detector_result[1]),
+            expected_trough_segment_results['min_value']
+        )
+
+    def test_get_bounds_for_segment_corner_cases(self):
+        detector = anomaly_detector.AnomalyDetector('test_id')
+        empty_segment = pd.Series([])
+        same_values_segment = pd.Series([2,2,2,2,2,2])
+        empty_detector_result = detector.get_bounds_for_segment(empty_segment)
+        same_values_detector_result = detector.get_bounds_for_segment(same_values_segment)
+
+        self.assertEqual(len(empty_detector_result[0]), 0)
+        self.assertEqual(len(empty_detector_result[1]), 0)
+        self.assertEqual(min(same_values_detector_result[0]), 0)
+        self.assertEqual(max(same_values_detector_result[0]), 0)
+        self.assertEqual(min(same_values_detector_result[1]), 0)
+        self.assertEqual(max(same_values_detector_result[1]), 0)
+
 if __name__ == '__main__':
     unittest.main()