def testIndependent(self): a = DictList([Object("o1"), Object("o2")]) b = DictList() self.assertIn("o1", a) self.assertNotIn("o1", b) b.append(Object("o3")) self.assertNotIn("o3", a) self.assertIn("o3", b)
def test_independent(self): a = DictList([Object("o1"), Object("o2")]) b = DictList() assert "o1" in a assert "o1" not in b b.append(Object("o3")) assert "o3" not in a assert "o3" in b
class TestDictList(TestCase): def setUp(self): self.obj = Object("test1") self.list = DictList() self.list.append(self.obj) def testAppend(self): obj2 = Object("test2") self.list.append(obj2) self.assertRaises(ValueError, self.list.append, Object("test1")) self.assertEqual(self.list.index(obj2), 1) self.assertEqual(self.list[1], obj2) self.assertEqual(self.list.get_by_id("test2"), obj2) self.assertEqual(len(self.list), 2) def testExtend(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list.extend(obj_list) self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testIadd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list += obj_list self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testAdd(self): from os import name as __name if __name == 'java': warn('\t\n**cobra.test.unit_tests.testAdd does not yet work with %s'%__name) return obj_list = [Object("test%d" % (i)) for i in range(2, 10)] sum = self.list + obj_list self.assertEqual(self.list[0].id, "test1") self.assertEqual(sum[1].id, "test2") self.assertEqual(sum.get_by_id("test2"), obj_list[0]) self.assertEqual(sum[8].id, "test9") self.assertEqual(len(self.list), 1) self.assertEqual(len(sum), 9) def testDeepcopy(self): from copy import deepcopy copied = deepcopy(self.list) for i, v in enumerate(self.list): assert self.list[i].id == copied[i].id assert self.list[i] is not copied[i]
class TestDictList(TestCase): def setUp(self): self.obj = Object("test1") self.list = DictList() self.list.append(self.obj) def testAppend(self): obj2 = Object("test2") self.list.append(obj2) self.assertRaises(ValueError, self.list.append, Object("test1")) self.assertEqual(self.list.index(obj2), 1) self.assertEqual(self.list[1], obj2) self.assertEqual(self.list.get_by_id("test2"), obj2) self.assertEqual(len(self.list), 2) def testExtend(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list.extend(obj_list) self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testIadd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list += obj_list self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testAdd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] sum = self.list + obj_list self.assertEqual(self.list[0].id, "test1") self.assertEqual(sum[1].id, "test2") self.assertEqual(sum.get_by_id("test2"), obj_list[0]) self.assertEqual(sum[8].id, "test9") self.assertEqual(len(self.list), 1) self.assertEqual(len(sum), 9)
class TestDictList(TestCase): def setUp(self): self.obj = Object("test1") self.list = DictList() self.list.append(self.obj) def testAppend(self): obj2 = Object("test2") self.list.append(obj2) self.assertRaises(ValueError, self.list.append, Object("test1")) self.assertEqual(self.list.index(obj2), 1) self.assertEqual(self.list[1], obj2) self.assertEqual(self.list.get_by_id("test2"), obj2) self.assertEqual(len(self.list), 2) def testExtend(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list.extend(obj_list) self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testIadd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list += obj_list self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testAdd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] sum = self.list + obj_list self.assertEqual(self.list[0].id, "test1") self.assertEqual(sum[1].id, "test2") self.assertEqual(sum.get_by_id("test2"), obj_list[0]) self.assertEqual(sum[8].id, "test9") self.assertEqual(len(self.list), 1) self.assertEqual(len(sum), 9) def testDeepcopy(self): from copy import deepcopy copied = deepcopy(self.list) for i, v in enumerate(self.list): assert self.list[i].id == copied[i].id assert self.list[i] is not copied[i] def testQuery(self): obj2 = Object("test2") self.list.append(obj2) result = self.list.query("test1") # matches only test1 self.assertEqual(len(result), 1) self.assertEqual(result[0], self.obj) result = self.list.query("test") # matches test1 and test2 self.assertEqual(len(result), 2)
class TestDictList(TestCase): def setUp(self): self.obj = Object("test1") self.list = DictList() self.list.append(self.obj) def testAppend(self): obj2 = Object("test2") self.list.append(obj2) self.assertRaises(ValueError, self.list.append, Object("test1")) self.assertEqual(self.list.index(obj2), 1) self.assertEqual(self.list[1], obj2) self.assertEqual(self.list.get_by_id("test2"), obj2) self.assertEqual(len(self.list), 2) def testExtend(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list.extend(obj_list) self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testIadd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list += obj_list self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testAdd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] sum = self.list + obj_list self.assertEqual(self.list[0].id, "test1") self.assertEqual(sum[1].id, "test2") self.assertEqual(sum.get_by_id("test2"), obj_list[0]) self.assertEqual(sum[8].id, "test9") self.assertEqual(len(self.list), 1) self.assertEqual(len(sum), 9) def testDeepcopy(self): from copy import deepcopy copied = deepcopy(self.list) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertIsNot(self.list[i], copied[i]) def testQuery(self): obj2 = Object("test2") self.list.append(obj2) result = self.list.query("test1") # matches only test1 self.assertEqual(len(result), 1) self.assertEqual(result[0], self.obj) result = self.list.query("test") # matches test1 and test2 self.assertEqual(len(result), 2)
def _build_targets(self): targets = DictList() for reaction in self.reactions: reaction = self._replace_adapted_metabolites(reaction) targets.append(ReactionKnockinTarget(reaction.id, reaction)) for reaction in self.exchanges: reaction = self._replace_adapted_metabolites(reaction) targets.append(ReactionKnockinTarget(reaction.id, reaction)) product = self._replace_adapted_metabolites(self.product) product.lower_bound = 0 targets.append(ReactionKnockinTarget(product.id, product)) return targets
def _build_targets(self): targets = DictList() for reaction in self.reactions: reaction = self._replace_adapted_metabolites(reaction) if reaction.id in metanetx.mnx2all: target = ReactionKnockinTarget(reaction.id, reaction, accession_id=reaction.id, accession_db='metanetx') else: target = ReactionKnockinTarget(reaction.id, reaction) targets.append(target) for reaction in self.exchanges: reaction = self._replace_adapted_metabolites(reaction) targets.append(ReactionKnockinTarget(reaction.id, reaction)) product = self._replace_adapted_metabolites(self.product) product.lower_bound = 0 targets.append(ReactionKnockinTarget(product.id, product)) return targets
class TestDictList(TestCase): def setUp(self): self.obj = Object("test1") self.list = DictList() self.list.append(self.obj) def testContains(self): self.assertIn(self.obj, self.list) self.assertIn(self.obj.id, self.list) self.assertNotIn(Object("not_in"), self.list) self.assertNotIn('not_in', self.list) def testIndex(self): self.assertEqual(self.list.index("test1"), 0) self.assertEqual(self.list.index(self.obj), 0) self.assertRaises(ValueError, self.list.index, "f") self.assertRaises(ValueError, self.list.index, Object("f")) def testIndependent(self): a = DictList([Object("o1"), Object("o2")]) b = DictList() self.assertIn("o1", a) self.assertNotIn("o1", b) b.append(Object("o3")) self.assertNotIn("o3", a) self.assertIn("o3", b) def testAppend(self): obj2 = Object("test2") self.list.append(obj2) self.assertRaises(ValueError, self.list.append, Object("test1")) self.assertEqual(self.list.index(obj2), 1) self.assertEqual(self.list[1], obj2) self.assertEqual(self.list.get_by_id("test2"), obj2) self.assertEqual(len(self.list), 2) def testExtend(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list.extend(obj_list) self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) self.assertRaises(ValueError, self.list.extend, [Object("test1")]) # Even if the object is unique, if it is present twice in the new # list, it should still raise an exception self.assertRaises(ValueError, self.list.extend, [Object("testd"), Object("testd")]) def testIadd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list += obj_list self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testAdd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] sum = self.list + obj_list self.assertIsNot(sum, self.list) self.assertIsNot(sum, obj_list) self.assertEqual(self.list[0].id, "test1") self.assertEqual(sum[1].id, "test2") self.assertEqual(sum.get_by_id("test2"), obj_list[0]) self.assertEqual(sum[8].id, "test9") self.assertEqual(len(self.list), 1) self.assertEqual(len(sum), 9) def testInitCopy(self): self.list.append(Object("test2")) copied = DictList(self.list) self.assertIsNot(self.list, copied) self.assertIsInstance(copied, self.list.__class__) self.assertEqual(len(self.list), len(copied)) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertEqual(i, copied.index(v.id)) self.assertIs(self.list[i], copied[i]) self.assertIs(v, copied.get_by_id(v.id)) def testSlice(self): self.list.append(Object("test2")) self.list.append(Object("test3")) sliced = self.list[:-1] self.assertIsNot(self.list, sliced) self.assertIsInstance(sliced, self.list.__class__) self.assertEqual(len(self.list), len(sliced) + 1) for i, v in enumerate(sliced): self.assertEqual(self.list[i].id, sliced[i].id) self.assertEqual(i, sliced.index(v.id)) self.assertIs(self.list[i], sliced[i]) self.assertIs(self.list[i], sliced.get_by_id(v.id)) def testCopy(self): self.list.append(Object("test2")) copied = copy(self.list) self.assertIsNot(self.list, copied) self.assertIsInstance(copied, self.list.__class__) self.assertEqual(len(self.list), len(copied)) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertEqual(i, copied.index(v.id)) self.assertIs(self.list[i], copied[i]) self.assertIs(v, copied.get_by_id(v.id)) def testDeepcopy(self): self.list.append(Object("test2")) copied = deepcopy(self.list) self.assertIsNot(self.list, copied) self.assertIsInstance(copied, self.list.__class__) self.assertEqual(len(self.list), len(copied)) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertEqual(i, copied.index(v.id)) self.assertIsNot(self.list[i], copied[i]) self.assertIsNot(v, copied.get_by_id(v.id)) def testPickle(self): self.list.append(Object("test2")) for protocol in range(HIGHEST_PROTOCOL): pickle_str = dumps(self.list, protocol=protocol) copied = loads(pickle_str) self.assertIsNot(self.list, copied) self.assertIsInstance(copied, self.list.__class__) self.assertEqual(len(self.list), len(copied)) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertEqual(i, copied.index(v.id)) self.assertIsNot(self.list[i], copied[i]) self.assertIsNot(v, copied.get_by_id(v.id)) def testQuery(self): obj2 = Object("test2") self.list.append(obj2) result = self.list.query("test1") # matches only test1 self.assertEqual(len(result), 1) self.assertEqual(result[0], self.obj) result = self.list.query("test") # matches test1 and test2 self.assertEqual(len(result), 2) def testRemoval(self): obj_list = DictList(Object("test%d" % (i)) for i in range(2, 10)) del obj_list[3] self.assertNotIn("test5", obj_list) self.assertEqual(obj_list.index(obj_list[-1]), len(obj_list) - 1) del obj_list[3:5] self.assertNotIn("test6", obj_list) self.assertNotIn("test7", obj_list) self.assertEqual(obj_list.index(obj_list[-1]), len(obj_list) - 1) removed = obj_list.pop(1) self.assertEqual(obj_list.index(obj_list[-1]), len(obj_list) - 1) self.assertEqual(removed.id, "test3") self.assertNotIn("test3", obj_list) def testSet(self): obj_list = DictList(Object("test%d" % (i)) for i in range(10)) obj_list[4] = Object("testa") self.assertEqual(obj_list.index("testa"), 4) self.assertEqual(obj_list[4].id, "testa") obj_list[5:7] = [Object("testb"), Object("testc")] self.assertEqual(obj_list.index("testb"), 5) self.assertEqual(obj_list[5].id, "testb") self.assertEqual(obj_list.index("testc"), 6) self.assertEqual(obj_list[6].id, "testc") # Even if the object is unique, if it is present twice in the new # list, it should still raise an exception self.assertRaises(ValueError, obj_list.__setitem__, slice(5, 7), [Object("testd"), Object("testd")])
def create_metanetx_universal_model(validate=False, verbose=False): """ Create an universal model from MetaNetX universal reactions and metabolites. The MetaNetX metabolite list is very large and includes metabolites that are not used in any reaction. The returned model only includes metabolites that are actually used in a reaction. Parameters ---------- validate : bool, optional When True, perform validity checks on universal COBRA model verbose : bool, optional When True, show warning messages Returns ------- cobra.Model COBRA model object with universal reactions and metabolites """ # Create an empty model. universal = Model('metanetx_universal', name='MetaNetX universal model') # Download the metabolites file. metabolite_list = _download_metanetx_file('chem_prop.tsv') # Map field names to column numbers (MetaNetX may add fields in future). field_names = { 'MNX_ID': 0, 'Description': 1, 'Formula': 2, 'Charge': 3, 'Mass': 4, 'InChI': 5, 'SMILES': 6, 'Source': 7, 'InChIKey': 8 } # Accumulate all available Metabolite objects separately. Later when creating # reactions, metabolites are put in a compartment. all_metabolites = DictList() # Create Metabolite objects for all of the metabolites from the downloaded file. # Add all of the universal metabolites from the list. LOGGER.info('Started creating Metabolite objects from %d lines in file', len(metabolite_list)) for index in range(len(metabolite_list)): if len(metabolite_list[index] ) == 0 or metabolite_list[index][0] == '#': continue # Skip empty lines and comment lines fields = metabolite_list[index].split('\t') if len(fields) < len(field_names): if verbose: warn('Skipped metabolite on line {0} with missing fields: {1}'. format(index, metabolite_list[index])) continue # Create cobra.core.Metabolite from MetaNetX metabolite. metabolite = Metabolite(id=fields[field_names['MNX_ID']], name=fields[field_names['Description']], formula=fields[field_names['Formula']]) charge = fields[field_names['Charge']] metabolite.charge = int( charge) if len(charge) > 0 and charge != 'NA' else None mass = fields[field_names['Mass']] if len(mass) > 0: metabolite.notes['mass'] = float(mass) metabolite.notes['InChI'] = fields[field_names['InChI']] \ if len(fields[field_names['InChI']]) > 0 else 'NA' metabolite.notes['SMILES'] = fields[field_names['SMILES']] \ if len(fields[field_names['SMILES']]) > 0 else 'NA' metabolite.notes['source'] = fields[field_names['Source']] \ if len(fields[field_names['Source']]) > 0 else 'NA' metabolite.notes['InChIKey'] = fields[field_names['InChIKey']] \ if len(fields[field_names['InChIKey']]) > 0 else 'NA' all_metabolites.append(metabolite) LOGGER.info('Finished creating %d Metabolite objects', len(all_metabolites)) # Download the compartments file. compartment_list = _download_metanetx_file('comp_prop.tsv') # Map field names to column numbers (MetaNetX may add fields in future). field_names = {'MNX_ID': 0, 'Description': 1, 'Source': 2} # Add the compartments to the universal model. LOGGER.info('Started adding compartments from %d lines in file', len(compartment_list)) for index in range(len(compartment_list)): if len(compartment_list[index] ) == 0 or compartment_list[index][0] == '#': continue # Skip empty lines and comment lines fields = compartment_list[index].split('\t') if len(fields) < len(field_names): if verbose: warn( 'Skipped compartment on line {0} with missing fields: {1}'. format(index, compartment_list[index])) continue universal.compartments[fields[field_names['MNX_ID']]] = fields[ field_names['Description']] LOGGER.info('Finished adding {0} compartments to universal model'.format( len(universal.compartments))) # Download the reactions file. reaction_list = _download_metanetx_file('reac_prop.tsv') # Map field names to column numbers (hopefully MetaNetX doesn't change this). field_names = { 'MNX_ID': 0, 'Equation': 1, 'Description': 2, 'Balance': 3, 'EC': 4, 'Source': 5 } # Accumulate Reaction and Metabolite objects separately because it is faster # than adding them one at a time to a model. reactions = DictList() metabolites = DictList() # Create Reaction objects for all of the reactions from the downloaded file. LOGGER.info('Started creating Reaction objects from %d lines in file', len(reaction_list)) for index in range(len(reaction_list)): if len(reaction_list[index]) == 0 or reaction_list[index][0] == '#': continue # Skip empty lines and comment lines fields = reaction_list[index].split('\t') if len(fields) != len(field_names): if verbose: warn('Skipped reaction on line {0} with missing fields: {1}'. format(index, reaction_list[index])) continue # Create cobra.core.Reaction from MetaNetX reaction. metabolite_info = _parse_metanetx_equation( fields[field_names['Equation']]) if metabolite_info is None: if verbose: warn( 'Could not parse equation for reaction {0} on line {1}: {2}' .format(fields[field_names['MNX_ID']], index, fields[field_names['Equation']])) continue rxn_mets = dict() for metabolite_id in metabolite_info: try: rxn_mets[metabolites.get_by_id( metabolite_id )] = metabolite_info[metabolite_id]['coefficient'] except KeyError: metabolite = all_metabolites.get_by_id( metabolite_info[metabolite_id]['mnx_id']).copy() metabolite.id = metabolite_id metabolite.compartment = metabolite_info[metabolite_id][ 'compartment'] metabolites.append(metabolite) rxn_mets[metabolite] = metabolite_info[metabolite_id][ 'coefficient'] reaction = Reaction(id=fields[field_names['MNX_ID']], name=fields[field_names['MNX_ID']], lower_bound=-1000.0, upper_bound=1000.0) reaction.add_metabolites(rxn_mets) if len(fields[field_names['EC']]) > 0: reaction.notes['EC_number'] = fields[field_names['EC']] if len(fields[field_names['Source']]) > 1: parts = fields[field_names['Source']].split(':') if len(parts) == 2: reaction.notes['aliases'] = {parts[0]: parts[1]} else: if verbose: warn('Could not parse source for {0}: {1}'.format( fields[field_names['MNX_ID']], fields[field_names['Source']])) reactions.append(reaction) LOGGER.info('Finished creating %d Reaction objects', len(reactions)) # Add the reactions to the universal model. universal.add_reactions(reactions) LOGGER.info('Finished adding Reaction objects to universal model') # If requested, validate the COBRA model. if validate: warn('Coming soon') return universal
class TestDictList(TestCase): def setUp(self): self.obj = Object("test1") self.list = DictList() self.list.append(self.obj) def testContains(self): self.assertIn(self.obj, self.list) self.assertIn(self.obj.id, self.list) self.assertNotIn(Object("not_in"), self.list) self.assertNotIn('not_in', self.list) def testIndex(self): self.assertEqual(self.list.index("test1"), 0) self.assertEqual(self.list.index(self.obj), 0) self.assertRaises(ValueError, self.list.index, "f") self.assertRaises(ValueError, self.list.index, Object("f")) # ensure trying to index with an object that is a different object # also raises an error self.assertRaises(ValueError, self.list.index, Object("test1")) def testIndependent(self): a = DictList([Object("o1"), Object("o2")]) b = DictList() self.assertIn("o1", a) self.assertNotIn("o1", b) b.append(Object("o3")) self.assertNotIn("o3", a) self.assertIn("o3", b) def testAppend(self): obj2 = Object("test2") self.list.append(obj2) self.assertRaises(ValueError, self.list.append, Object("test1")) self.assertEqual(self.list.index(obj2), 1) self.assertEqual(self.list[1], obj2) self.assertIs(self.list.get_by_id("test2"), obj2) self.assertEqual(len(self.list), 2) def testInsert(self): obj2 = Object("a") self.list.insert(0, obj2) self.assertEqual(self.list.index(obj2), 0) self.assertEqual(self.list.index("test1"), 1) self.assertIs(self.list.get_by_id("a"), obj2) self.assertEqual(len(self.list), 2) self.assertRaises(ValueError, self.list.append, obj2) def testExtend(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list.extend(obj_list) self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) self.assertRaises(ValueError, self.list.extend, [Object("test1")]) # Even if the object is unique, if it is present twice in the new # list, it should still raise an exception self.assertRaises(ValueError, self.list.extend, [Object("testd"), Object("testd")]) def testIadd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list += obj_list self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testAdd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] sum = self.list + obj_list self.assertIsNot(sum, self.list) self.assertIsNot(sum, obj_list) self.assertEqual(self.list[0].id, "test1") self.assertEqual(sum[1].id, "test2") self.assertEqual(sum.get_by_id("test2"), obj_list[0]) self.assertEqual(sum[8].id, "test9") self.assertEqual(len(self.list), 1) self.assertEqual(len(sum), 9) def testInitCopy(self): self.list.append(Object("test2")) copied = DictList(self.list) self.assertIsNot(self.list, copied) self.assertIsInstance(copied, self.list.__class__) self.assertEqual(len(self.list), len(copied)) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertEqual(i, copied.index(v.id)) self.assertIs(self.list[i], copied[i]) self.assertIs(v, copied.get_by_id(v.id)) def testSlice(self): self.list.append(Object("test2")) self.list.append(Object("test3")) sliced = self.list[:-1] self.assertIsNot(self.list, sliced) self.assertIsInstance(sliced, self.list.__class__) self.assertEqual(len(self.list), len(sliced) + 1) for i, v in enumerate(sliced): self.assertEqual(self.list[i].id, sliced[i].id) self.assertEqual(i, sliced.index(v.id)) self.assertIs(self.list[i], sliced[i]) self.assertIs(self.list[i], sliced.get_by_id(v.id)) def testCopy(self): self.list.append(Object("test2")) copied = copy(self.list) self.assertIsNot(self.list, copied) self.assertIsInstance(copied, self.list.__class__) self.assertEqual(len(self.list), len(copied)) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertEqual(i, copied.index(v.id)) self.assertIs(self.list[i], copied[i]) self.assertIs(v, copied.get_by_id(v.id)) def testDeepcopy(self): self.list.append(Object("test2")) copied = deepcopy(self.list) self.assertIsNot(self.list, copied) self.assertIsInstance(copied, self.list.__class__) self.assertEqual(len(self.list), len(copied)) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertEqual(i, copied.index(v.id)) self.assertIsNot(self.list[i], copied[i]) self.assertIsNot(v, copied.get_by_id(v.id)) def testPickle(self): self.list.append(Object("test2")) for protocol in range(HIGHEST_PROTOCOL): pickle_str = dumps(self.list, protocol=protocol) copied = loads(pickle_str) self.assertIsNot(self.list, copied) self.assertIsInstance(copied, self.list.__class__) self.assertEqual(len(self.list), len(copied)) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertEqual(i, copied.index(v.id)) self.assertIsNot(self.list[i], copied[i]) self.assertIsNot(v, copied.get_by_id(v.id)) def testQuery(self): obj2 = Object("test2") obj2.name = "foobar1" self.list.append(obj2) result = self.list.query("test1") # matches only test1 self.assertEqual(len(result), 1) self.assertEqual(result[0], self.obj) result = self.list.query("foo", "name") # matches only test2 self.assertEqual(len(result), 1) self.assertEqual(result[0], obj2) result = self.list.query("test") # matches test1 and test2 self.assertEqual(len(result), 2) # test with a regular expression result = self.list.query(re.compile("test[0-9]")) self.assertEqual(len(result), 2) result = self.list.query(re.compile("test[29]")) self.assertEqual(len(result), 1) # test query of name result = self.list.query(re.compile("foobar."), "name") self.assertEqual(len(result), 1) def testRemoval(self): obj_list = DictList(Object("test%d" % (i)) for i in range(2, 10)) del obj_list[3] self.assertNotIn("test5", obj_list) self.assertEqual(obj_list.index(obj_list[-1]), len(obj_list) - 1) self.assertEqual(len(obj_list), 7) del obj_list[3:5] self.assertNotIn("test6", obj_list) self.assertNotIn("test7", obj_list) self.assertEqual(obj_list.index(obj_list[-1]), len(obj_list) - 1) self.assertEqual(len(obj_list), 5) removed = obj_list.pop(1) self.assertEqual(obj_list.index(obj_list[-1]), len(obj_list) - 1) self.assertEqual(removed.id, "test3") self.assertNotIn("test3", obj_list) self.assertEqual(len(obj_list), 4) removed = obj_list.pop() self.assertEqual(removed.id, "test9") self.assertNotIn(removed.id, obj_list) self.assertEqual(len(obj_list), 3) def testSet(self): obj_list = DictList(Object("test%d" % (i)) for i in range(10)) obj_list[4] = Object("testa") self.assertEqual(obj_list.index("testa"), 4) self.assertEqual(obj_list[4].id, "testa") obj_list[5:7] = [Object("testb"), Object("testc")] self.assertEqual(obj_list.index("testb"), 5) self.assertEqual(obj_list[5].id, "testb") self.assertEqual(obj_list.index("testc"), 6) self.assertEqual(obj_list[6].id, "testc") # Even if the object is unique, if it is present twice in the new # list, it should still raise an exception self.assertRaises(ValueError, obj_list.__setitem__, slice(5, 7), [Object("testd"), Object("testd")]) def testSortandReverse(self): dl = DictList(Object("test%d" % (i)) for i in reversed(range(10))) self.assertEqual(dl[0].id, "test9") dl.sort() self.assertEqual(len(dl), 10) self.assertEqual(dl[0].id, "test0") self.assertEqual(dl.index("test0"), 0) dl.reverse() self.assertEqual(dl[0].id, "test9") self.assertEqual(dl.index("test0"), 9) def testDir(self): """makes sure tab complete will work""" attrs = dir(self.list) self.assertIn("test1", attrs) self.assertIn("_dict", attrs) # attribute of DictList def testUnion(self): self.list.union([Object("test1"), Object("test2")]) # should only add 1 element self.assertEqual(len(self.list), 2) self.assertEqual(self.list.index("test2"), 1)
class TestDictList(TestCase): def setUp(self): self.obj = Object("test1") self.list = DictList() self.list.append(self.obj) def testIndependent(self): a = DictList([Object("o1"), Object("o2")]) b = DictList() self.assertIn("o1", a) self.assertNotIn("o1", b) b.append(Object("o3")) self.assertNotIn("o3", a) self.assertIn("o3", b) def testAppend(self): obj2 = Object("test2") self.list.append(obj2) self.assertRaises(ValueError, self.list.append, Object("test1")) self.assertEqual(self.list.index(obj2), 1) self.assertEqual(self.list[1], obj2) self.assertEqual(self.list.get_by_id("test2"), obj2) self.assertEqual(len(self.list), 2) def testExtend(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list.extend(obj_list) self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testIadd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] self.list += obj_list self.assertEqual(self.list[1].id, "test2") self.assertEqual(self.list.get_by_id("test2"), obj_list[0]) self.assertEqual(self.list[8].id, "test9") self.assertEqual(len(self.list), 9) def testAdd(self): obj_list = [Object("test%d" % (i)) for i in range(2, 10)] sum = self.list + obj_list self.assertEqual(self.list[0].id, "test1") self.assertEqual(sum[1].id, "test2") self.assertEqual(sum.get_by_id("test2"), obj_list[0]) self.assertEqual(sum[8].id, "test9") self.assertEqual(len(self.list), 1) self.assertEqual(len(sum), 9) def testDeepcopy(self): from copy import deepcopy copied = deepcopy(self.list) for i, v in enumerate(self.list): self.assertEqual(self.list[i].id, copied[i].id) self.assertIsNot(self.list[i], copied[i]) def testQuery(self): obj2 = Object("test2") self.list.append(obj2) result = self.list.query("test1") # matches only test1 self.assertEqual(len(result), 1) self.assertEqual(result[0], self.obj) result = self.list.query("test") # matches test1 and test2 self.assertEqual(len(result), 2) def testRemoval(self): obj_list = DictList(Object("test%d" % (i)) for i in range(2, 10)) del obj_list[3] self.assertNotIn("test5", obj_list) self.assertEqual(obj_list.index(obj_list[-1]), len(obj_list) - 1) del obj_list[3:5] self.assertNotIn("test6", obj_list) self.assertNotIn("test7", obj_list) self.assertEqual(obj_list.index(obj_list[-1]), len(obj_list) - 1) removed = obj_list.pop(1) self.assertEqual(obj_list.index(obj_list[-1]), len(obj_list) - 1) self.assertEqual(removed.id, "test3") self.assertNotIn("test3", obj_list) def testSet(self): obj_list = DictList(Object("test%d" % (i)) for i in range(10)) obj_list[4] = Object("testa") self.assertEqual(obj_list.index("testa"), 4) self.assertEqual(obj_list[4].id, "testa") obj_list[5:7] = [Object("testb"), Object("testc")] self.assertEqual(obj_list.index("testb"), 5) self.assertEqual(obj_list[5].id, "testb") self.assertEqual(obj_list.index("testc"), 6) self.assertEqual(obj_list[6].id, "testc")
def dict_list(): obj = Object("test1") test_list = DictList() test_list.append(obj) return obj, test_list