def test_apply_to_attributes_delegates_to_correct_function_when_passed_dict( mocker): mocker.patch.object(graph_operations, 'apply_mapping_to_attributes') graph_operations.apply_to_attributes([1, 2, 3], { 1: 1, 2: 2, 3: 3 }, 'location') graph_operations.apply_mapping_to_attributes.assert_called_once_with( [1, 2, 3], { 1: 1, 2: 2, 3: 3 }, 'location')
def test_applying_function_to_attributes(): d = graph_operations.apply_to_attributes([(1, { 'a': 2, 'b': 4 }), (2, { '1': 4, '2': 8 }), (3, { 'a': 6, 'b': 10 })], add_attributes, 'location') assert_semantically_equal(d, {1: {'location': 6}, 3: {'location': 16}})
def test_applying_dict_map_to_attributes(): d = graph_operations.apply_to_attributes([(1, { 'a': 2, 'b': 4 }), (2, { 'a': 4, 'b': 8 }), (3, { 'a': 6, 'b': 10 })], { 2: 0, 4: 10 }, 'a') assert_semantically_equal(d, {1: {'a': 0}, 2: {'a': 10}})
def test_apply_to_attributes_delegates_to_correct_function_when_passed_callable( mocker): mocker.patch.object(graph_operations, 'apply_function_to_attributes') graph_operations.apply_to_attributes([1, 2, 3], do_nothing, 'location') graph_operations.apply_function_to_attributes.assert_called_once_with( [1, 2, 3], do_nothing, 'location')