def test_extend_maybe_dict(self): list1 = {1: [2]} extend_maybe(list1, None) self.assertEqual({1: [2]}, list1) extend_maybe(list1, {}) self.assertEqual({1: [2]}, list1) extend_maybe(list1, {3: [4]}) self.assertEqual({1: [2], 3: [4]}, list1) extend_maybe(list1, {1: [5]}) self.assertEqual({1: [2, 5], 3: [4]}, list1)
def test_extend_maybe_list(self): list1 = [1] extend_maybe(list1, None) self.assertEqual([1], list1) extend_maybe(list1, []) self.assertEqual([1], list1) extend_maybe(list1, [2]) self.assertEqual([1, 2], list1)