def test_drop(self):
''' drop. '''
lst = [str(lpe) for lpe in range(le.NUM)]
for loops in self._gen_loop_combs():
ddls = DataDimLoops(*loops)
sublst = ddls.drop(lst)
self.assertEqual(len(sublst), le.NUM - len(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))