def test_valid_repeated_args(self):
''' Valid repeated arguments. '''
ddls = DataDimLoops(le.IFM, le.OFM, le.IFM, le.IFM)
self.assertTupleEqual(ddls.loops(), (le.IFM, le.OFM))
ddls = DataDimLoops(*([le.BAT] * 10))
self.assertTupleEqual(ddls.loops(), (le.BAT, ))
def test_valid_args(self):
''' Valid arguments. '''
ddls = DataDimLoops(le.IFM, le.OFM)
self.assertTupleEqual(ddls.loops(), (le.IFM, le.OFM))
ddls = DataDimLoops(le.BAT, le.IFM, le.OFM)
self.assertTupleEqual(ddls.loops(), (le.IFM, le.OFM, le.BAT))
def test_data_cnt(self):
''' Get data_cnt. '''
lcnt = [3, 5, 7]
for loops in self._gen_loop_combs():
ddls = DataDimLoops(*loops)
dcnt = ddls.data_cnt(lcnt)
dcnt2 = 1
for lpe in ddls.loops():
dcnt2 *= lcnt[lpe]
self.assertEqual(dcnt, dcnt2)
def test_loops(self):
''' Get loops. '''
for loops in self._gen_loop_combs():
ddls = DataDimLoops(*loops)
self.assertTupleEqual(ddls.loops(), loops)
