def test_get_indices_empty_result(): positions = create_grid() inds = get_indices_for_pos(positions, .1, 3.5, .5, .9) assert(inds.size == 0)
def test_get_indices_empty_result(): positions = create_grid() inds = get_indices_for_pos(positions, .1, 3.5, .5, .9) assert (inds.size == 0)
def test_get_indices(): positions = create_grid() inds = get_indices_for_pos(positions, .1, 3.5, .5, 3) inds_gt = np.array([6,7,8,11,12,13,16,17,18]) assert(inds.all() == inds_gt.all())
def test_get_indices(): positions = create_grid() inds = get_indices_for_pos(positions, .1, 3.5, .5, 3) inds_gt = np.array([6, 7, 8, 11, 12, 13, 16, 17, 18]) assert (inds.all() == inds_gt.all())