comparison tests/test_signal.py @ 331:9741b52f194a

Reformat code and documentation

 class TestSignal(TestCase):
     def setUp(self):
         self.data = b"012345679ABC"
         self.numpy_fmt = {"b": np.int8, "h": np.int16, "i": np.int32}
 
-
     @genty_dataset(
         int8_mono=(1, [48, 49, 50, 51, 52, 53, 54, 55, 57, 65, 66, 67]),
         int16_mono=(2, [12592, 13106, 13620, 14134, 16697, 17218]),
         int32_mono=(4, [858927408, 926299444, 1128415545]),
-        int8_stereo=(1,  [[48, 50, 52, 54, 57, 66], [49, 51, 53, 55, 65, 67]]),
+        int8_stereo=(1, [[48, 50, 52, 54, 57, 66], [49, 51, 53, 55, 65, 67]]),
         int16_stereo=(2, [[12592, 13620, 16697], [13106, 14134, 17218]]),
         int32_3channel=(4, [[858927408], [926299444], [1128415545]]),
     )
     def test_to_array(self, sample_width, expected):
         if isinstance(expected[0], list):
             channels = len(expected)
-            expected = [array_(signal_.FORMAT[sample_width], xi) for xi in expected]
+            expected = [
+                array_(signal_.FORMAT[sample_width], xi) for xi in expected
+            ]
         else:
             channels = 1
             expected = array_(signal_.FORMAT[sample_width], expected)
         resutl = signal_.to_array(self.data, sample_width, channels)
         resutl_numpy = signal_numpy.to_array(self.data, sample_width, channels)

         energy = signal_.calculate_energy_single_channel(x, sample_width)
         self.assertEqual(energy, expected)
         energy = signal_numpy.calculate_energy_single_channel(x, sample_width)
         self.assertEqual(energy, expected)
 
-
     @genty_dataset(
         min_=(
             [[300, 320, 400, 600], [150, 160, 200, 300]],
             2,
             min,

             2,
             max,
             52.50624901923348,
         ),
     )
-    def test_calculate_energy_multichannel(self, x, sample_width, aggregation_fn, expected):
+    def test_calculate_energy_multichannel(
+        self, x, sample_width, aggregation_fn, expected
+    ):
         x = [array_(signal_.FORMAT[sample_width], xi) for xi in x]
-        energy = signal_.calculate_energy_multichannel(x, sample_width, aggregation_fn)
+        energy = signal_.calculate_energy_multichannel(
+            x, sample_width, aggregation_fn
+        )
         self.assertEqual(energy, expected)
 
-        energy = signal_numpy.calculate_energy_multichannel(x, sample_width, aggregation_fn)
+        energy = signal_numpy.calculate_energy_multichannel(
+            x, sample_width, aggregation_fn
+        )
         self.assertEqual(energy, expected)
+
 
 if __name__ == "__main__":
     unittest.main()