1
0
mirror of synced 2025-12-19 17:37:05 -05:00

fix tests

This commit is contained in:
Sam Bentley
2022-09-16 22:44:57 +01:00
parent 6ea9dbcf42
commit e48642dd73

View File

@@ -7,10 +7,10 @@ import numpy as np
class TestQvdReader(): class TestQvdReader():
def test_read_shape(self): def test_read_shape(self):
qvd = qvd_reader.read(f'{os.path.dirname(__file__)}/test_files/AAPL.qvd') qvd = qvd_reader.read(f'{os.path.dirname(__file__)}/test_files/AAPL.qvd')
csv = pd.read_csv('test_files/AAPL.csv', float_precision='round_trip') csv = pd.read_csv(f'{os.path.dirname(__file__)}/test_files/AAPL.csv', float_precision='round_trip')
assert qvd.shape == csv.shape assert qvd.shape == csv.shape
def test_read_size(self): def test_read_size(self):
qvd = qvd_reader.read(f'{os.path.dirname(__file__)}/test_files/AAPL.qvd') qvd = qvd_reader.read(f'{os.path.dirname(__file__)}/test_files/AAPL.qvd')
csv = pd.read_csv('test_files/AAPL.csv', float_precision='round_trip') csv = pd.read_csv(f'{os.path.dirname(__file__)}/test_files/AAPL.csv', float_precision='round_trip')
assert np.array_equal(np.sort(qvd.columns, axis=0), np.sort(csv.columns, axis=0)) assert np.array_equal(np.sort(qvd.columns, axis=0), np.sort(csv.columns, axis=0))