Update test_tf_base.py

Author: kujaku11

tests/core/transfer_function/test_tf_base.py

@@ -41,9 +41,7 @@ def test_shape_zeros_dtype(self):
                 self.assertEqual((tf.values == 0).all(), v_dict["empty"])
 
     def test_frequency(self):
-        self.assertEqual(
-            (self.tf.frequency == 1.0 / np.arange(1, 3, 1)).all(), True
-        )
+        self.assertEqual((self.tf.frequency == 1.0 / np.arange(1, 3, 1)).all(), True)
 
     def test_period(self):
         self.assertEqual((self.tf.period == np.arange(1, 3, 1)).all(), True)
@@ -55,9 +53,7 @@ def test_equal(self):
 class TestTFBaseTFErrorInput(unittest.TestCase):
     @classmethod
     def setUpClass(self):
-        self.tf = TFBase(
-            tf_error=np.array([[[0, 1], [1, 0]], [[1, 0], [0, 1]]])
-        )
+        self.tf = TFBase(tf_error=np.array([[[0, 1], [1, 0]], [[1, 0], [0, 1]]]))
         self.expected_shape = (2, 2, 2)
         self.expected = {
             "transfer_function": {"dtype": complex, "empty": True},
@@ -78,9 +74,7 @@ def test_shape_zeros_dtype(self):
                     self.assertEqual((tf.values == 0).all(), v_dict["empty"])
 
     def test_frequency(self):
-        self.assertEqual(
-            (self.tf.frequency == 1.0 / np.arange(1, 3, 1)).all(), True
-        )
+        self.assertEqual((self.tf.frequency == 1.0 / np.arange(1, 3, 1)).all(), True)
 
     def test_period(self):
         self.assertEqual((self.tf.period == np.arange(1, 3, 1)).all(), True)
@@ -89,9 +83,7 @@ def test_period(self):
 class TestTFBaseTFModelErrorInput(unittest.TestCase):
     @classmethod
     def setUpClass(self):
-        self.tf = TFBase(
-            tf_model_error=np.array([[[0, 1], [1, 0]], [[1, 0], [0, 1]]])
-        )
+        self.tf = TFBase(tf_model_error=np.array([[[0, 1], [1, 0]], [[1, 0], [0, 1]]]))
         self.expected_shape = (2, 1, 1)
         self.expected = {
             "transfer_function": {"dtype": complex, "empty": True},
@@ -112,9 +104,7 @@ def test_shape_zeros_dtype(self):
                     self.assertEqual((tf.values == 0).all(), v_dict["empty"])
 
     def test_frequency(self):
-        self.assertEqual(
-            (self.tf.frequency == 1.0 / np.arange(1, 3, 1)).all(), True
-        )
+        self.assertEqual((self.tf.frequency == 1.0 / np.arange(1, 3, 1)).all(), True)
 
     def test_period(self):
         self.assertEqual((self.tf.period == np.arange(1, 3, 1)).all(), True)
@@ -320,9 +310,7 @@ def setUpClass(self):
         self.new_periods = np.logspace(-3, 3, 12)
 
     def interpolate(self, interp_type, bounds_error=False):
-        interp_tf = spi.interp1d(
-            self.period, self.tf, axis=0, kind=interp_type
-        )
+        interp_tf = spi.interp1d(self.period, self.tf, axis=0, kind=interp_type)
         interp_tf_error = spi.interp1d(
             self.period, self.tf_error, axis=0, kind=interp_type
         )
@@ -437,9 +425,7 @@ def setUpClass(self):
             frequency=1.0 / self.period,
         )
 
-        self.tf_interpolated_same_period = self.tf_base.interpolate(
-            self.period
-        )
+        self.tf_interpolated_same_period = self.tf_base.interpolate(self.period)
 
         self.new_period = np.logspace(-4, 4, 24)
         self.tf_interpolated_different_period = self.tf_base.interpolate(
@@ -471,9 +457,7 @@ def test_backfill_nans(self):
         )
         self.assertEqual(
             True,
-            np.all(
-                np.isclose(np.nan_to_num(self.tf), np.nan_to_num(new_tf_nans))
-            ),
+            np.all(np.isclose(np.nan_to_num(self.tf), np.nan_to_num(new_tf_nans))),
         )
 
     def test_same_tf(self):
@@ -494,9 +478,7 @@ def test_same_tf_error(self):
             True,
             np.all(
                 np.isclose(
-                    np.nan_to_num(
-                        self.tf_base._dataset.transfer_function_error
-                    ),
+                    np.nan_to_num(self.tf_base._dataset.transfer_function_error),
                     np.nan_to_num(
                         self.tf_interpolated_same_period._dataset.transfer_function_error
                     ),
@@ -509,9 +491,7 @@ def test_same_tf_model_error(self):
             True,
             np.all(
                 np.isclose(
-                    np.nan_to_num(
-                        self.tf_base._dataset.transfer_function_model_error
-                    ),
+                    np.nan_to_num(self.tf_base._dataset.transfer_function_model_error),
                     np.nan_to_num(
                         self.tf_interpolated_same_period._dataset.transfer_function_model_error
                     ),
@@ -532,15 +512,11 @@ def test_different_period_index(self):
             with self.subTest(
                 f"period_min: in = {true_entry['input']} out = {true_entry['output']}"
             ):
-                self.assertTrue(
-                    true_entry["period_min"] <= new_entry["period_min"]
-                )
+                self.assertTrue(true_entry["period_min"] <= new_entry["period_min"])
             with self.subTest(
                 f"period_max: in = {true_entry['input']} out = {true_entry['output']}"
             ):
-                self.assertTrue(
-                    true_entry["period_max"] >= new_entry["period_max"]
-                )
+                self.assertTrue(true_entry["period_max"] >= new_entry["period_max"])
 
 
 # =============================================================================