def test_init_with_args(self): synonym_obj = Synonym('synonym') assert synonym_obj is not None assert synonym_obj.name == 'synonym' self.add_objs_and_assert_ids(synonym_obj)
def test_add_synonym_to_oil(self): oil_obj = ImportedRecord(**OilTestCase.get_mock_oil_file_record()) synonym_obj = Synonym('test oil') oil_obj.synonyms.append(synonym_obj) self.add_objs_and_assert_ids([oil_obj, synonym_obj]) assert oil_obj.synonyms == [synonym_obj] assert synonym_obj.imported == [oil_obj]
def add_synonyms(session, oil, row_dict): if row_dict.get('Synonyms'): for s in row_dict.get('Synonyms').split(','): s = s.strip() if len(s) > 0: synonyms = (session.query(Synonym).filter( Synonym.name == s).all()) if len(synonyms) > 0: # we link the existing synonym object oil.synonyms.append(synonyms[0]) else: # we add a new synonym object oil.synonyms.append(Synonym(s))
def test_oils_that_share_a_synonym(self): oil_args = OilTestCase.get_mock_oil_file_record() oil_obj1 = ImportedRecord(**oil_args) oil_args.update({u'oil_name': u'Test Oil 2', u'adios_oil_id': u'AD99998'}) oil_obj2 = ImportedRecord(**oil_args) synonym_obj = Synonym('test oil') oil_obj1.synonyms.append(synonym_obj) oil_obj2.synonyms.append(synonym_obj) self.add_objs_and_assert_ids([oil_obj1, oil_obj2, synonym_obj]) assert oil_obj1.synonyms == [synonym_obj] assert oil_obj2.synonyms == [synonym_obj] assert synonym_obj.imported == [oil_obj1, oil_obj2]
def test_init_no_args(self): with raises(TypeError): synonym_obj = Synonym() assert synonym_obj is not None