def test_skip(self):
for skip in [0, 3, 13]:
r1 = DataInMemory(self.d)
out_with_skip = r1.get_output(skip=skip)[0]
r2 = DataInMemory(self.d)
out = r2.get_output()[0]
np.testing.assert_almost_equal(out_with_skip, out[skip::],
err_msg="The first %s rows were skipped, but that did not "
"match the rows with skip=0 and sliced by [%s::]" % (skip, skip))
def test_skip_input_list(self):
for skip in [0, 3, 13]:
r1 = DataInMemory([self.d, self.d])
out_with_skip = r1.get_output(skip=skip)
r2 = DataInMemory([self.d, self.d])
out = r2.get_output()
np.testing.assert_almost_equal(out_with_skip[0], out[0][skip::],
err_msg="The first %s rows of the first file were skipped, but that did not "
"match the rows with skip=0 and sliced by [%s::]" % (skip, skip))
np.testing.assert_almost_equal(out_with_skip[1], out[1][skip::],
err_msg="The first %s rows of the second file were skipped, but that did not"
" match the rows with skip=0 and sliced by [%s::]" % (skip, skip))
def test_stride(self):
reader = DataInMemory(self.d)
stride = [1, 2, 3, 4, 5, 6, 7, 10, 11, 21, 23]
for s in stride:
output = reader.get_output(stride=s)[0]
expected = self.d[::s]
np.testing.assert_allclose(output, expected,
err_msg="not equal for stride=%i" % s)
