diff tests/test_signal.py @ 331:9741b52f194a

Reformat code and documentation

--- a/tests/test_signal.py	Wed Oct 23 21:24:33 2019 +0100
+++ b/tests/test_signal.py	Thu Oct 24 20:49:51 2019 +0200
@@ -13,19 +13,20 @@
         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)
@@ -146,7 +147,6 @@
         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]],
@@ -161,13 +161,20 @@
             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()
\ No newline at end of file
+    unittest.main()