fix tests
This commit is contained in:
@@ -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))
|
||||||
|
|||||||
Reference in New Issue
Block a user