def test_rotation(self, idcode, times): cols = self._cols(times) body = Body(idcode) data = body.rotation(times) assert len(data) == 2 assert len(data[0]) == cols assert data[0].dtype == float for matrix in data[1]: assert matrix.shape == (3, 3) assert matrix.dtype == float