def test_take(self): ''' take. ''' lst = [str(lpe) for lpe in range(le.NUM)] for loops in self._gen_loop_combs(): ddls = DataDimLoops(*loops) sublst = ddls.take(lst) self.assertEqual(len(sublst), len(loops)) self.assertListEqual(sublst, [str(lpe) for lpe in loops])
def test_take_and_drop(self): ''' take and drop. ''' lst = [str(lpe) for lpe in range(le.NUM)] for loops in self._gen_loop_combs(): ddls = DataDimLoops(*loops) takelst = ddls.take(lst) droplst = ddls.drop(lst) self.assertEqual(len(takelst) + len(droplst), le.NUM) self.assertTrue(set(takelst).isdisjoint(set(droplst))) self.assertSetEqual(set(takelst) | set(droplst), set(lst))