def test_tuple_iterator(self):
data_map = MultiDataMap(self.new_style_map)
data_map.iterator = MultiDataMap.TupleIterator
tuples = [item for item in data_map]
self.assertEqual(len(tuples), 4)
self.assertTrue(all(isinstance(item, tuple) for item in tuples))
self.assertTrue(all(len(item) == 2 for item in tuples))
self.assertEqual(tuples[0], ('locus001', ['L12345_SB101.MS']))
def test_skip_iterator(self):
data_map = MultiDataMap(self.new_style_map)
data_map.iterator = MultiDataMap.SkipIterator
unskipped = [item for item in data_map]
self.assertEqual(len(unskipped), 2)
self.assertTrue(all(isinstance(item, MultiDataProduct) for item in unskipped))
self.assertEqual(unskipped[0].host, 'locus002')
self.assertEqual(unskipped[0].file, ['L12345_SB102.MS'])
def test_tuple_iterator(self):
data_map = MultiDataMap(self.new_style_map)
data_map.iterator = MultiDataMap.TupleIterator
tuples = [item for item in data_map]
self.assertEqual(len(tuples), 4)
self.assertTrue(all(isinstance(item, tuple) for item in tuples))
self.assertTrue(all(len(item) == 2 for item in tuples))
self.assertEqual(tuples[0], ('locus001', ['L12345_SB101.MS']))
def test_skip_iterator(self):
data_map = MultiDataMap(self.new_style_map)
data_map.iterator = MultiDataMap.SkipIterator
unskipped = [item for item in data_map]
self.assertEqual(len(unskipped), 2)
self.assertTrue(
all(isinstance(item, MultiDataProduct) for item in unskipped))
self.assertEqual(unskipped[0].host, 'locus002')
self.assertEqual(unskipped[0].file, ['L12345_SB102.MS'])
def test_append_item_non_skip(self):
data_map = MultiDataMap(self.new_style_map)
data_map.append(("host", ["file"], False, [False]))
data_map.append(("host", ["file"], False))
data_map.iterator = DataMap.TupleIterator
tuples = [item for item in data_map]
self.assertEqual(len(tuples), 6)
self.assertTrue(all(isinstance(item, tuple) for item in tuples))
self.assertTrue(all(len(item) == 2 for item in tuples))
self.assertEqual(tuples[-1], ('host', ['file']))
def test_append_item_non_skip(self):
data_map = MultiDataMap(self.new_style_map)
data_map.append(("host", ["file"], False, [False] ))
data_map.append(("host", ["file"], False))
data_map.iterator = DataMap.TupleIterator
tuples = [item for item in data_map]
self.assertEqual(len(tuples), 6)
self.assertTrue(all(isinstance(item, tuple) for item in tuples))
self.assertTrue(all(len(item) == 2 for item in tuples))
self.assertEqual(tuples[-1], ('host', ['file']))
def test_append_item_skip(self):
data_map = MultiDataMap(self.new_style_map)
data_map.append(("host", ["file"], True, [True]))
data_map.append(("host", ["file"], True))
data_map.iterator = DataMap.SkipIterator
dataProducts = [item for item in data_map]
# default contains 2 nonskipped items
self.assertEqual(len(dataProducts), 2)
self.assertTrue(
all(isinstance(item, MultiDataProduct) for item in dataProducts))
# The map already contains 2 skipped items, the final item is tested
# here
self.assertEqual(dataProducts[-1].host, 'locus004')
self.assertEqual(dataProducts[-1].file, ['L12345_SB104.MS'])
def test_append_item_skip(self):
data_map = MultiDataMap(self.new_style_map)
data_map.append(("host",["file"], True, [True]))
data_map.append(("host",["file"], True))
data_map.iterator = DataMap.SkipIterator
dataProducts = [item for item in data_map]
# default contains 2 nonskipped items
self.assertEqual(len(dataProducts), 2)
self.assertTrue(all(isinstance(item, MultiDataProduct)
for item in dataProducts))
# The map already contains 2 skipped items, the final item is tested
# here
self.assertEqual(dataProducts[-1].host, 'locus004')
self.assertEqual(dataProducts[-1].file, ['L12345_SB104.MS'])
