init

.venv/lib/python3.8/site-packages/pygame/tests/sndarray_test.py

@@ -0,0 +1,155 @@
+import unittest
+
+from numpy import int8, int16, uint8, uint16, float32, array, alltrue
+
+import pygame
+import pygame.sndarray
+
+
+class SndarrayTest(unittest.TestCase):
+    array_dtypes = {8: uint8, -8: int8, 16: uint16, -16: int16, 32: float32}
+
+    def _assert_compatible(self, arr, size):
+        dtype = self.array_dtypes[size]
+        self.assertEqual(arr.dtype, dtype)
+
+    def test_array(self):
+        def check_array(size, channels, test_data):
+            try:
+                pygame.mixer.init(22050, size, channels, allowedchanges=0)
+            except pygame.error:
+                # Not all sizes are supported on all systems.
+                return
+            try:
+                __, sz, __ = pygame.mixer.get_init()
+                if sz == size:
+                    srcarr = array(test_data, self.array_dtypes[size])
+                    snd = pygame.sndarray.make_sound(srcarr)
+                    arr = pygame.sndarray.array(snd)
+                    self._assert_compatible(arr, size)
+                    self.assertTrue(
+                        alltrue(arr == srcarr),
+                        "size: %i\n%s\n%s" % (size, arr, test_data),
+                    )
+            finally:
+                pygame.mixer.quit()
+
+        check_array(8, 1, [0, 0x0F, 0xF0, 0xFF])
+        check_array(8, 2, [[0, 0x80], [0x2D, 0x41], [0x64, 0xA1], [0xFF, 0x40]])
+        check_array(16, 1, [0, 0x00FF, 0xFF00, 0xFFFF])
+        check_array(
+            16, 2, [[0, 0xFFFF], [0xFFFF, 0], [0x00FF, 0xFF00], [0x0F0F, 0xF0F0]]
+        )
+        check_array(-8, 1, [0, -0x80, 0x7F, 0x64])
+        check_array(-8, 2, [[0, -0x80], [-0x64, 0x64], [0x25, -0x50], [0xFF, 0]])
+        check_array(-16, 1, [0, 0x7FFF, -0x7FFF, -1])
+        check_array(-16, 2, [[0, -0x7FFF], [-0x7FFF, 0], [0x7FFF, 0], [0, 0x7FFF]])
+
+    def test_get_arraytype(self):
+        array_type = pygame.sndarray.get_arraytype()
+
+        self.assertEqual(array_type, "numpy", "unknown array type %s" % array_type)
+
+    def test_get_arraytypes(self):
+        arraytypes = pygame.sndarray.get_arraytypes()
+        self.assertIn("numpy", arraytypes)
+
+        for atype in arraytypes:
+            self.assertEqual(atype, "numpy", "unknown array type %s" % atype)
+
+    def test_make_sound(self):
+        def check_sound(size, channels, test_data):
+            try:
+                pygame.mixer.init(22050, size, channels, allowedchanges=0)
+            except pygame.error:
+                # Not all sizes are supported on all systems.
+                return
+            try:
+                __, sz, __ = pygame.mixer.get_init()
+                if sz == size:
+                    srcarr = array(test_data, self.array_dtypes[size])
+                    snd = pygame.sndarray.make_sound(srcarr)
+                    arr = pygame.sndarray.samples(snd)
+                    self.assertTrue(
+                        alltrue(arr == srcarr),
+                        "size: %i\n%s\n%s" % (size, arr, test_data),
+                    )
+            finally:
+                pygame.mixer.quit()
+
+        check_sound(8, 1, [0, 0x0F, 0xF0, 0xFF])
+        check_sound(8, 2, [[0, 0x80], [0x2D, 0x41], [0x64, 0xA1], [0xFF, 0x40]])
+        check_sound(16, 1, [0, 0x00FF, 0xFF00, 0xFFFF])
+        check_sound(
+            16, 2, [[0, 0xFFFF], [0xFFFF, 0], [0x00FF, 0xFF00], [0x0F0F, 0xF0F0]]
+        )
+        check_sound(-8, 1, [0, -0x80, 0x7F, 0x64])
+        check_sound(-8, 2, [[0, -0x80], [-0x64, 0x64], [0x25, -0x50], [0xFF, 0]])
+        check_sound(-16, 1, [0, 0x7FFF, -0x7FFF, -1])
+        check_sound(-16, 2, [[0, -0x7FFF], [-0x7FFF, 0], [0x7FFF, 0], [0, 0x7FFF]])
+        check_sound(32, 2, [[0.0, -1.0], [-1.0, 0], [1.0, 0], [0, 1.0]])
+
+    def test_samples(self):
+
+        null_byte = b"\x00"
+
+        def check_sample(size, channels, test_data):
+            try:
+                pygame.mixer.init(22050, size, channels, allowedchanges=0)
+            except pygame.error:
+                # Not all sizes are supported on all systems.
+                return
+            try:
+                __, sz, __ = pygame.mixer.get_init()
+                if sz == size:
+                    zeroed = null_byte * ((abs(size) // 8) * len(test_data) * channels)
+                    snd = pygame.mixer.Sound(buffer=zeroed)
+                    samples = pygame.sndarray.samples(snd)
+                    self._assert_compatible(samples, size)
+                    ##print ('X %s' % (samples.shape,))
+                    ##print ('Y %s' % (test_data,))
+                    samples[...] = test_data
+                    arr = pygame.sndarray.array(snd)
+                    self.assertTrue(
+                        alltrue(samples == arr),
+                        "size: %i\n%s\n%s" % (size, arr, test_data),
+                    )
+            finally:
+                pygame.mixer.quit()
+
+        check_sample(8, 1, [0, 0x0F, 0xF0, 0xFF])
+        check_sample(8, 2, [[0, 0x80], [0x2D, 0x41], [0x64, 0xA1], [0xFF, 0x40]])
+        check_sample(16, 1, [0, 0x00FF, 0xFF00, 0xFFFF])
+        check_sample(
+            16, 2, [[0, 0xFFFF], [0xFFFF, 0], [0x00FF, 0xFF00], [0x0F0F, 0xF0F0]]
+        )
+        check_sample(-8, 1, [0, -0x80, 0x7F, 0x64])
+        check_sample(-8, 2, [[0, -0x80], [-0x64, 0x64], [0x25, -0x50], [0xFF, 0]])
+        check_sample(-16, 1, [0, 0x7FFF, -0x7FFF, -1])
+        check_sample(-16, 2, [[0, -0x7FFF], [-0x7FFF, 0], [0x7FFF, 0], [0, 0x7FFF]])
+        check_sample(32, 2, [[0.0, -1.0], [-1.0, 0], [1.0, 0], [0, 1.0]])
+
+    def test_use_arraytype(self):
+        def do_use_arraytype(atype):
+            pygame.sndarray.use_arraytype(atype)
+
+        pygame.sndarray.use_arraytype("numpy")
+        self.assertEqual(pygame.sndarray.get_arraytype(), "numpy")
+
+        self.assertRaises(ValueError, do_use_arraytype, "not an option")
+
+    def test_float32(self):
+        """sized arrays work with Sounds and 32bit float arrays."""
+        try:
+            pygame.mixer.init(22050, 32, 2, allowedchanges=0)
+        except pygame.error:
+            # Not all sizes are supported on all systems.
+            self.skipTest("unsupported mixer configuration")
+
+        arr = array([[0.0, -1.0], [-1.0, 0], [1.0, 0], [0, 1.0]], float32)
+        newsound = pygame.mixer.Sound(array=arr)
+        pygame.mixer.quit()
+
+
+if __name__ == "__main__":
+    unittest.main()