def test_fix_missing_tgids(self): with contextlib.nested(open(ATRACE_EXTRACTED_TGIDS, 'r'), open(ATRACE_MISSING_TGIDS, 'r'), open(ATRACE_FIXED_TGIDS, 'r')) as (f1, f2, f3): atrace_data = f2.read() tgid_map = eval(f1.read()) fixed = f3.read() res = atrace_agent.fix_missing_tgids(atrace_data, tgid_map) self.assertEqual(res, fixed)