def test_get_group_ixs_conversion_to_bools():
group_ids = range(4)
ixs = utils.get_group_ixs(group_ids, bools=True)
expected_ix = {
0: np.array([True, False, False, False]),
1: np.array([False, True, False, False]),
2: np.array([False, False, True, False]),
3: np.array([False, False, False, True]),
}
assert_dicts_are_equal(ixs, expected_ix)
def test_get_group_ixs_matrix_of_ids():
group_ids = np.array([[0, 1], [1, 0], [1, 0], [1, 0]])
ixs = utils.get_group_ixs(group_ids)
expected_ix = {(0, 1): [0], (1, 0): [1, 2, 3]}
assert_dicts_are_equal(ixs, expected_ix)
def test_get_group_ixs_list_of_list_of_ids():
group_ids = [[0, 1], [1, 0], [1, 1], [1, 0]]
ixs = utils.get_group_ixs(group_ids)
expected_ix = {(0, 1): [0], (1, 0): [1, 3], (1, 1): [2]}
assert_dicts_are_equal(ixs, expected_ix)
def test_get_group_ixs_multiple_id_args():
group0_ids = [0, 1, 2, 0]
group1_ids = [1, 0, 2, 1]
ixs = utils.get_group_ixs(group0_ids, group1_ids)
expected_ix = {(0, 1): [0, 3], (1, 0): [1], (2, 2): [2]}
assert_dicts_are_equal(ixs, expected_ix)
def test_get_group_ixs_vector_of_ids():
group_ids = np.array([0, 1, 0, 1])
ixs = utils.get_group_ixs(group_ids)
expected_ix = {0: [0, 2], 1: [1, 3]}
assert_dicts_are_equal(ixs, expected_ix)
def test_get_group_ixs_list_of_ids():
group_ids = [0, 1, 0, 2]
ixs = utils.get_group_ixs(group_ids)
expected_ix = {0: [0, 2], 1: [1], 2: [3]}
assert_dicts_are_equal(ixs, expected_ix)