def test_coerce_to_dict_with_list_of_lists(): test_input = [[0, 0, 1], [0, 1, 0]] expected_output = { '[0, 0, 1]': [0, 0, 1], '[0, 1, 0]': [0, 1, 0], } actual_output = glmr._coerce_to_dict(test_input) assert actual_output == expected_output
def test_coerce_to_dict_with_list_of_strings(): test_input = ['contrast_name_0', 'contrast_name_1'] expected_output = { 'contrast_name_0': 'contrast_name_0', 'contrast_name_1': 'contrast_name_1', } actual_output = glmr._coerce_to_dict(test_input) assert actual_output == expected_output
def test_coerce_to_dict_with_list_of_ints(): test_input = [1, 0, 1] expected_output = {'[1, 0, 1]': [1, 0, 1]} actual_output = glmr._coerce_to_dict(test_input) assert np.array_equal( actual_output['[1, 0, 1]'], expected_output['[1, 0, 1]'], )
def test_coerce_to_dict_with_array_of_ints(): test_input = np.array([1, 0, 1]) expected_output = {'[1 0 1]': np.array([1, 0, 1])} actual_output = glmr._coerce_to_dict(test_input) assert expected_output.keys() == actual_output.keys() assert np.array_equal( actual_output['[1 0 1]'], expected_output['[1 0 1]'], )
def test_coerce_to_dict_with_dict(): test_input = { 'contrast_0': [0, 0, 1], 'contrast_1': [0, 1, 1], } expected_output = { 'contrast_0': [0, 0, 1], 'contrast_1': [0, 1, 1], } actual_output = glmr._coerce_to_dict(test_input) assert actual_output == expected_output
def test_coerce_to_dict_with_list_of_arrays(): test_input = [np.array([0, 0, 1]), np.array([0, 1, 0])] expected_output = { '[0 0 1]': np.array([0, 0, 1]), '[0 1 0]': np.array([0, 1, 0]), } actual_output = glmr._coerce_to_dict(test_input) assert actual_output.keys() == expected_output.keys() for key in actual_output: assert np.array_equal( actual_output[key], expected_output[key], )
def test_coerce_to_dict_with_string(): test_input = 'StopSuccess - Go' expected_output = {'StopSuccess - Go': 'StopSuccess - Go'} actual_output = glmr._coerce_to_dict(test_input) assert actual_output == expected_output