def test_read_raw_compressed(): input_file = 'test.tsv.gz' with gzip.open(input_file, 'wt') as g: g.write(test_input_data) raw = reader.read_raw(input_file) os.remove(input_file) assert_frame_equal(raw, expected_raw)
def test_read_raw(): input_file = 'test.tsv' with open(input_file, 'wt') as f: f.write(test_input_data) raw = reader.read_raw(input_file) os.remove(input_file) assert_frame_equal(raw, expected_raw)