def test_grid_coords1D(ndarray_type):
df_copy = df.copy()
xlim, ylim = traja.trajectory._get_xylim(df_copy)
bins = traja.trajectory._bins_to_tuple(df_copy, None)
grid_indices = traja.grid_coordinates(df_copy,
bins=bins,
xlim=xlim,
ylim=ylim)
if ndarray_type:
grid_indices = grid_indices.values
grid_indices1D = traja._grid_coords1D(grid_indices)
assert isinstance(grid_indices1D, np.ndarray)
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)