def test_usecols(self):
reader = NumPyFileReader(self.f4)
cols = (0, 2)
it = reader.iterator(chunk=0, return_trajindex=False, cols=cols)
with it:
for x in it:
np.testing.assert_equal(x, self.d2[:, cols])
def test_lagged_stridden_access(self):
reader = NumPyFileReader(self.f1)
strides = [2, 3, 5, 7, 15]
lags = [1, 3, 7, 10, 30]
for stride in strides:
for lag in lags:
chunks = []
for _, _, Y in reader.iterator(stride, lag):
chunks.append(Y)
chunks = np.vstack(chunks)
np.testing.assert_equal(chunks, self.d[lag::stride])
def test_lagged_stridden_access(self):
reader = NumPyFileReader(self.f1)
strides = [2, 3, 5, 7, 15]
lags = [1, 3, 7, 10, 30]
for stride in strides:
for lag in lags:
chunks = []
for _, _, Y in reader.iterator(stride, lag):
chunks.append(Y)
chunks = np.vstack(chunks)
np.testing.assert_equal(chunks, self.d[lag::stride])
def test_lagged_stridden_access_multiple_files(self):
reader = NumPyFileReader(self.files2d)
strides = [2, 3, 5, 7, 15]
lags = [1, 3, 7, 10, 30]
for stride in strides:
for lag in lags:
chunks = {
i: []
for i in range(reader.number_of_trajectories())
}
for itraj, _, Y in reader.iterator(stride, lag):
chunks[itraj].append(Y)
for i, k in enumerate(chunks.values()):
stack = np.vstack(k)
d = np.load(self.files2d[i])
np.testing.assert_equal(
stack, d[lag::stride], "not equal for stride=%i"
" and lag=%i" % (stride, lag))
