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)