def test_imap_log(self): data = [ { "action": "action_0", "data": "data_0" }, { "action": "action_1", "data": "data_1" }, ] f = self.to_file_like(data) def f_action_0(item): return ("action_0", item["data"]) def f_action_1(item): return ("action_1", item["data"]) res_iter = reader.imap_log(reader.read(f), { "action_0": f_action_0, "action_1": f_action_1 }) self.assertEquals([("action_0", "data_0"), ("action_1", "data_1")], list(res_iter))
def test_imap_log(self): data = [{"action": "action_0", "data": "data_0"}, {"action": "action_1", "data": "data_1"}] f = self.to_file_like(data) def f_action_0(item): return ("action_0", item["data"]) def f_action_1(item): return ("action_1", item["data"]) res_iter = reader.imap_log(reader.read(f), {"action_0": f_action_0, "action_1": f_action_1}) self.assertEquals([("action_0", "data_0"), ("action_1", "data_1")], list(res_iter))