def test_position(self, idcode, times): cols = self._cols(times) body = Body(idcode) data = body.position(times) assert data.shape == (4, cols) assert data.dtype == float