def test_example(args): name, cmd, baseline_file = args ext = os.path.splitext(baseline_file)[1][1:] scrub = SCRUBBERS[ext] assert_equals = ASSERT_EQUALS.get(ext, None) startdir = os.getcwd() os.chdir(datadir) output = scrub(check_output(cmd).decode()) with open(baseline_file) as f: baseline = scrub(f.read()) if assert_equals is not None: assert_equals(output, baseline) else: assert output == baseline os.chdir(startdir)
def test_example(example): cmd = example.script baseline_file = example.baseline ext = os.path.splitext(baseline_file)[1][1:] scrub = SCRUBBERS[ext] assert_equals = ASSERT_EQUALS.get(ext, None) startdir = os.getcwd() os.chdir(datadir) output = scrub(subprocess.check_output(cmd).decode()) with open(baseline_file) as f: baseline = scrub(f.read()) if assert_equals is not None: assert_equals(output, baseline) else: assert output == baseline os.chdir(startdir)