Polish of updated batch

Author: viblo

pymunk/batch.py

@@ -74,6 +74,7 @@
 ]
 
 from enum import Flag
+from typing import Any
 
 from ._chipmunk_cffi import ffi, lib
 from .space import Space
@@ -167,7 +168,7 @@ def float_buf(self) -> ffi.buffer:
             self._float_arr.arr, ffi.sizeof("cpFloat") * self._float_arr.num
         )
 
-    def set_float_buf(self, buffer) -> None:
+    def set_float_buf(self, buffer: Any) -> None:
         """Set the floating point internal data to the supplied buffer.
 
         buffer should be an array of floats, implmenting the buffer/memoryview
@@ -192,7 +193,7 @@ def int_buf(self) -> ffi.buffer:
             self._int_arr.arr, ffi.sizeof("uintptr_t") * self._int_arr.num
         )
 
-    def set_int_buf(self, buffer) -> None:
+    def set_int_buf(self, buffer: Any) -> None:
         """Set the integer internal data to the supplied buffer.
 
         buffer should be an array of ints, implmenting the buffer/memoryview

pymunk/tests/test_batch.py

@@ -8,7 +8,6 @@
 class UnitTestBatch(unittest.TestCase):
 
     def test_buffer(self) -> None:
-        print()
         b = pymunk.batch.Buffer()
         arr = array.array("d", [1.2, 3.4])
         b.set_float_buf(arr)
@@ -161,27 +160,6 @@ def test_set_bodies(self) -> None:
 
         self.assertEqual(list(memoryview(data.float_buf()).cast("d")), v)
 
-    def test_set_get_bodies(self) -> None:
-        s = pymunk.Space()
-
-        b1 = pymunk.Body(1, 1)
-        s.add(b1, pymunk.Circle(b1, 4))
-
-        b2 = pymunk.Body(1, 1)
-        s.add(b2, pymunk.Circle(b2, 4))
-
-        data = pymunk.batch.Buffer()
-        angle_arr = array.array("d", [3, 4, 5, 6, 7, 8])
-        data.set_float_buf(angle_arr)
-        pymunk.batch.set_space_bodies(
-            s, pymunk.batch.BodyFields.POSITION | pymunk.batch.BodyFields.ANGLE, data
-        )
-
-        data = pymunk.batch.Buffer()
-        pymunk.batch.get_space_bodies(
-            s, pymunk.batch.BodyFields.POSITION | pymunk.batch.BodyFields.ANGLE, data
-        )
-
     def test_get_arbiters(self) -> None:
         s = pymunk.Space()
 
@@ -275,7 +253,7 @@ def check_arb_data(arb: pymunk.Arbiter) -> None:
         b2.each_arbiter(check_arb_data)
         b3.each_arbiter(check_arb_data)
 
-    def test_get_arbiter_sensor(self):
+    def test_get_arbiter_sensor(self) -> None:
         s = pymunk.Space()
 
         b1 = pymunk.Body(1, 1)