def test_state(self, idcode, times): cols = self._cols(times) body = Body(idcode) data = body.state(times) assert data.shape == (8, cols) assert data.dtype == float