new test cases for TestExtractStateInterval

Author: R7L208

python/tests/tsdf_tests.py

@@ -455,7 +455,7 @@ def test_upsample(self):
 class ExtractStateIntervalsTest(SparkTest):
     """Test of finding time ranges for metrics with constant state."""
 
-    def test_eq_0_extractStateIntervals(self):
+    def test_eq_0(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -472,7 +472,7 @@ def test_eq_0_extractStateIntervals(self):
         self.assertDataFramesEqual(intervals_eq_1_df, expected_df)
         self.assertDataFramesEqual(intervals_eq_2_df, expected_df)
 
-    def test_eq_1_extractStateIntervals(self):
+    def test_eq_1(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -489,7 +489,7 @@ def test_eq_1_extractStateIntervals(self):
         self.assertDataFramesEqual(intervals_eq_1_df, expected_df)
         self.assertDataFramesEqual(intervals_eq_2_df, expected_df)
 
-    def test_ne_0_extractStateIntervals(self):
+    def test_ne_0(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -506,7 +506,7 @@ def test_ne_0_extractStateIntervals(self):
         self.assertDataFramesEqual(intervals_ne_0_df, expected_df)
         self.assertDataFramesEqual(intervals_ne_1_df, expected_df)
 
-    def test_ne_1_extractStateIntervals(self):
+    def test_ne_1(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -523,7 +523,7 @@ def test_ne_1_extractStateIntervals(self):
         self.assertDataFramesEqual(intervals_ne_0_df, expected_df)
         self.assertDataFramesEqual(intervals_ne_1_df, expected_df)
 
-    def test_gt_0_extractStateIntervals(self):
+    def test_gt_0(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -535,7 +535,7 @@ def test_gt_0_extractStateIntervals(self):
 
         self.assertDataFramesEqual(intervals_gt_df, expected_df)
 
-    def test_gt_1_extractStateIntervals(self):
+    def test_gt_1(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -547,7 +547,7 @@ def test_gt_1_extractStateIntervals(self):
 
         self.assertDataFramesEqual(intervals_gt_df, expected_df)
 
-    def test_lt_0_extractStateIntervals(self):
+    def test_lt_0(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -560,7 +560,7 @@ def test_lt_0_extractStateIntervals(self):
         # test extractStateIntervals_tsdf summary
         self.assertDataFramesEqual(intervals_lt_df, expected_df)
 
-    def test_lt_1_extractStateIntervals(self):
+    def test_lt_1(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -573,7 +573,7 @@ def test_lt_1_extractStateIntervals(self):
         # test intervals_tsdf summary
         self.assertDataFramesEqual(intervals_lt_df, expected_df)
 
-    def test_gte_0_extractStateIntervals(self):
+    def test_gte_0(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -585,7 +585,7 @@ def test_gte_0_extractStateIntervals(self):
 
         self.assertDataFramesEqual(intervals_gt_df, expected_df)
 
-    def test_gte_1_extractStateIntervals(self):
+    def test_gte_1(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -597,7 +597,7 @@ def test_gte_1_extractStateIntervals(self):
 
         self.assertDataFramesEqual(intervals_gt_df, expected_df)
 
-    def test_lte_0_extractStateIntervals(self):
+    def test_lte_0(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -610,7 +610,7 @@ def test_lte_0_extractStateIntervals(self):
         # test intervals_tsdf summary
         self.assertDataFramesEqual(intervals_lte_df, expected_df)
 
-    def test_lte_1_extractStateIntervals(self):
+    def test_lte_1(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -623,7 +623,7 @@ def test_lte_1_extractStateIntervals(self):
         # test extractStateIntervals_tsdf summary
         self.assertDataFramesEqual(intervals_lte_df, expected_df)
 
-    def test_threshold_fn_extractStateIntervals(self):
+    def test_threshold_fn(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -640,7 +640,7 @@ def threshold_fn(a: Column, b: Column) -> Column:
         # test extractStateIntervals_tsdf summary
         self.assertDataFramesEqual(extracted_intervals_df, expected_df)
 
-    def test_null_safe_eq_0_extractStateIntervals(self):
+    def test_null_safe_eq_0(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -652,7 +652,7 @@ def test_null_safe_eq_0_extractStateIntervals(self):
         # test extractStateIntervals_tsdf summary
         self.assertDataFramesEqual(intervals_eq_df, expected_df)
 
-    def test_null_safe_eq_1_extractStateIntervals(self):
+    def test_null_safe_eq_1(self):
         # construct dataframes
         input_tsdf: TSDF = self.get_data_as_tsdf("input")
         expected_df: DataFrame = self.get_data_as_sdf("expected")
@@ -664,6 +664,40 @@ def test_null_safe_eq_1_extractStateIntervals(self):
         # test extractStateIntervals_tsdf summary
         self.assertDataFramesEqual(intervals_eq_df, expected_df)
 
+    def test_adjacent_intervals(self):
+        # construct dataframes
+        input_tsdf: TSDF = self.get_data_as_tsdf("input")
+        expected_df: DataFrame = self.get_data_as_sdf("expected")
+
+        intervals_eq_df: DataFrame = input_tsdf.extractStateIntervals(
+            "metric_1", "metric_2", "metric_3"
+        )
+
+        # test extractStateIntervals_tsdf summary
+        self.assertDataFramesEqual(intervals_eq_df, expected_df)
+
+    def test_invalid_state_definition_str(self):
+        # construct dataframes
+        input_tsdf: TSDF = self.get_data_as_tsdf("input")
+
+        try:
+            input_tsdf.extractStateIntervals(
+                "metric_1", "metric_2", "metric_3", state_definition="N/A"
+            )
+        except ValueError as e:
+            self.assertEqual(type(e), ValueError)
+
+    def test_invalid_state_definition_type(self):
+        # construct dataframes
+        input_tsdf: TSDF = self.get_data_as_tsdf("input")
+
+        try:
+            input_tsdf.extractStateIntervals(
+                "metric_1", "metric_2", "metric_3", state_definition=0
+            )
+        except TypeError as e:
+            self.assertEqual(type(e), TypeError)
+
 
 # MAIN
 if __name__ == "__main__":