def grid_coordinates(self, **kwargs): return traja.grid_coordinates(self._obj, **kwargs)
def test_transition_matrix(): df_copy = df.copy() grid_indices = traja.grid_coordinates(df_copy) assert grid_indices.shape[1] == 2 grid_indices1D = traja._grid_coords1D(grid_indices) transitions_matrix = traja.transition_matrix(grid_indices1D)