def test_init_no_args(self):
oil_obj = ImportedRecord()
assert oil_obj is not None
assert oil_obj.id is None
with raises(IntegrityError):
self.session.add(oil_obj)
self.session.flush()
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_add_cut_to_oil(self):
oil_obj = ImportedRecord(**OilTestCase.get_mock_oil_file_record())
cut_obj = Cut(**CutTestCase.get_mock_cut_file_record())
oil_obj.cuts.append(cut_obj)
self.add_objs_and_assert_ids([oil_obj, cut_obj])
assert oil_obj.cuts == [cut_obj]
assert cut_obj.imported == oil_obj
def test_add_dvis_to_oil(self):
oil_obj = ImportedRecord(**OilTestCase.get_mock_oil_file_record())
dvis_obj = DVis(**DVisTestCase.get_mock_dvis_file_record())
oil_obj.dvis.append(dvis_obj)
self.add_objs_and_assert_ids([oil_obj, dvis_obj])
assert oil_obj.dvis == [dvis_obj]
assert dvis_obj.imported == oil_obj
def test_add_kvis_to_oil(self):
oil_obj = ImportedRecord(**OilTestCase.get_mock_oil_file_record())
kvis_obj = KVis(**KVisTestCase.get_mock_kvis_file_record())
oil_obj.kvis.append(kvis_obj)
self.add_objs_and_assert_ids([oil_obj, kvis_obj])
assert oil_obj.kvis == [kvis_obj]
assert kvis_obj.imported == oil_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 test_add_toxicity_to_oil(self):
oil_obj = ImportedRecord(**OilTestCase.get_mock_oil_file_record())
toxicity_obj = \
Toxicity(**ToxicityTestCase.get_mock_toxicity_file_record())
oil_obj.toxicities.append(toxicity_obj)
self.add_objs_and_assert_ids([oil_obj, toxicity_obj])
assert oil_obj.toxicities == [toxicity_obj]
assert toxicity_obj.imported == oil_obj
def test_add_density_to_oil(self):
oil_obj = ImportedRecord(**OilTestCase.get_mock_oil_file_record())
density_obj = \
Density(**DensityTestCase.get_mock_density_file_record())
oil_obj.densities.append(density_obj)
self.add_objs_and_assert_ids([oil_obj, density_obj])
assert oil_obj.densities == [density_obj]
assert density_obj.imported == oil_obj
def add_oil_object(session, file_columns, row_data):
file_columns = [slugify_filename(c).lower() for c in file_columns]
row_dict = dict(zip(file_columns, row_data))
fix_name(row_dict)
fix_pour_point(row_dict)
fix_flash_point(row_dict)
fix_preferred_oils(row_dict)
oil = ImportedRecord(**row_dict)
add_synonyms(session, oil, row_dict)
add_densities(oil, row_dict)
add_kinematic_viscosities(oil, row_dict)
add_dynamic_viscosities(oil, row_dict)
add_distillation_cuts(oil, row_dict)
add_toxicity_effective_concentrations(oil, row_dict)
add_toxicity_lethal_concentrations(oil, row_dict)
session.add(oil)
transaction.commit()
def test_init_with_args(self):
oil_obj = ImportedRecord(**self.get_mock_oil_file_record())
self.assert_mock_oil_object(oil_obj)
self.add_objs_and_assert_ids(oil_obj)