def test_get_2_elements_complex(self): k = 2 key_fn_2 = lambda x: x["arg2"] result_2_list = get_topk(self.complex_list, k, key_fn_2) self.assertListEqual(result_2_list, [self.item5, self.item4]) key_fn_1 = lambda x: x["arg1"] result_1_list = get_topk(self.complex_list, k, key_fn_1) self.assertListEqual(result_1_list, [self.item4, self.item3])
def test_get_3_elements_simple(self): k = 3 result_list = get_topk(self.simple_list, k, self.identity_fn) self.assertListEqual(result_list, [5, 4, 3])
def test_get_more_elements_than_size_simple(self): k = len(self.simple_list) + 1 result_list = get_topk(self.simple_list, k, self.identity_fn) self.assertListEqual(result_list, sorted(self.simple_list, reverse=True))