def test_DataChunkIterator_ignore_undetermined_axis(self):
# Compare data chunk iterators with undetermined axis (ignore axis)
d1 = DataChunkIterator(data=np.arange(10).reshape(2, 5))
d1.max_shape = (None, 5)
d2 = DataChunkIterator(data=np.arange(10).reshape(2, 5))
res = ShapeValidator.assertEqualShape(d1, d2, ignore_undetermined=True)
self.assertTrue(res.result)
self.assertIsNone(res.error)
self.assertTupleEqual(res.ignored, ((0, 0), ))
self.assertTupleEqual(res.unmatched, ())
self.assertTupleEqual(res.shape1, (None, 5))
self.assertTupleEqual(res.shape2, (2, 5))
self.assertTupleEqual(res.axes1, (0, 1))
self.assertTupleEqual(res.axes2, (0, 1))
def test_DataChunkIterator_error_on_undetermined_axis(self):
# Compare data chunk iterators with undetermined axis (error on undetermined axis)
d1 = DataChunkIterator(data=np.arange(10).reshape(2, 5))
d1.max_shape = (None, 5)
d2 = DataChunkIterator(data=np.arange(10).reshape(2, 5))
res = ShapeValidator.assertEqualShape(d1,
d2,
ignore_undetermined=False)
self.assertFalse(res.result)
self.assertEquals(res.error, 'AXIS_LEN_ERROR')
self.assertTupleEqual(res.ignored, ())
self.assertTupleEqual(res.unmatched, ((0, 0), ))
self.assertTupleEqual(res.shape1, (None, 5))
self.assertTupleEqual(res.shape2, (2, 5))
self.assertTupleEqual(res.axes1, (0, 1))
self.assertTupleEqual(res.axes2, (0, 1))