def test_get_oil(search, isNone): if isNone: with raises(NoResultFound): o = get_oil(search) else: o = get_oil(search) if isinstance(search, basestring): assert o.name == search
def test_get_oil_id(adios_id, name): if name is None: # shouldn't be found with raises(NoResultFound): o = get_oil(adios_id) else: o = get_oil(adios_id) assert o.name == name assert o.adios_oil_id == adios_id.strip().upper()
def test_get_oil_from_json(): ''' Ok, here we will test our ability to construct an oil object from a json payload. ''' oil_from_db = get_oil('ALASKA NORTH SLOPE (MIDDLE PIPELINE, 1997)') oil_from_json = Oil.from_json(oil_from_db.tojson()) # Do we want to fill in these properties??? assert oil_from_json.imported is None assert oil_from_json.estimated is None for attr in ('adhesion_kg_m_2', 'api', 'bullwinkle_fraction', 'bullwinkle_time', 'emulsion_water_fraction_max', 'flash_point_max_k', 'flash_point_min_k', 'id', 'name', 'oil_seawater_interfacial_tension_n_m', 'oil_seawater_interfacial_tension_ref_temp_k', 'oil_water_interfacial_tension_n_m', 'oil_water_interfacial_tension_ref_temp_k', 'pour_point_max_k', 'pour_point_min_k', 'solubility', 'sulphur_fraction', 'k0y',): assert getattr(oil_from_db, attr) == getattr(oil_from_json, attr) for db_obj, json_obj in zip(oil_from_db.cuts, oil_from_json.cuts): for attr in ('liquid_temp_k', 'vapor_temp_k', 'fraction'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.densities, oil_from_json.densities): for attr in ('kg_m_3', 'ref_temp_k', 'weathering'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.kvis, oil_from_json.kvis): for attr in ('m_2_s', 'ref_temp_k', 'weathering'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.molecular_weights, oil_from_json.molecular_weights): for attr in ('g_mol', 'ref_temp_k', 'sara_type'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.sara_densities, oil_from_json.sara_densities): for attr in ('density', 'ref_temp_k', 'sara_type'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.sara_fractions, oil_from_json.sara_fractions): for attr in ('fraction', 'ref_temp_k', 'sara_type'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.categories, oil_from_json.categories): assert getattr(db_obj, 'name') == getattr(json_obj, 'name') assert (getattr(db_obj.parent, 'name') == getattr(json_obj.parent, 'name'))
''' test functions in utilities modules ''' import numpy as np import pytest from oil_library import get_oil from oil_library.utilities import (get_density, get_viscosity, get_pour_point, get_boiling_points_from_api) oil_ = get_oil('LUCKENBACH FUEL OIL') # Test case - get ref temps from densities then append ref_temp for # density at 0th index for a few more values: # density_test = [d.ref_temp_k for d in oil_.densities] # density_test.append(oil_.densities[0].ref_temp_k) density_tests = [ oil_.densities[ix].ref_temp_k if ix < len(oil_.densities) else oil_.densities[0].ref_temp_k for ix in range(0, len(oil_.densities) + 3) ] density_exp = [ d.kg_m_3 for temp in density_tests for d in oil_.densities if abs(d.ref_temp_k - temp) == 0 ] ''' test get_density for - scalar - list, tuple - numpy arrays as row/column and with/without output arrays
''' This is just a scratchpad script I use inside ipython ''' import oil_library from oil_library import get_oil, _get_db_session from oil_library.models import Oil from oil_library.oil_props import OilProps from pprint import PrettyPrinter pp = PrettyPrinter(indent=2) session = _get_db_session() print 'Our session:', session.connection().engine # oil_obj = get_oil('LUCKENBACH FUEL OIL') oil_obj = get_oil('BAHIA') props_obj = OilProps(oil_obj) pp.pprint([f for f in props_obj.component_mass_fractions()]) oil_obj = session.query(Oil).filter(Oil.adios_oil_id == 'AD00584').one() my_str = oil_obj.imported.comments print my_str benz = get_oil(oil_library.sample_oils.benzene.json_data) print benz
def test_get_oil_from_json(): ''' Ok, here we will test our ability to construct an oil object from a json payload. ''' oil_from_db = get_oil('ALASKA NORTH SLOPE (MIDDLE PIPELINE, 1997)') oil_from_json = Oil.from_json(oil_from_db.tojson()) # Do we want to fill in these properties??? assert oil_from_json.imported is None assert oil_from_json.estimated is None for attr in ( 'adhesion_kg_m_2', 'api', 'bullwinkle_fraction', 'bullwinkle_time', 'emulsion_water_fraction_max', 'flash_point_max_k', 'flash_point_min_k', 'id', 'name', 'oil_seawater_interfacial_tension_n_m', 'oil_seawater_interfacial_tension_ref_temp_k', 'oil_water_interfacial_tension_n_m', 'oil_water_interfacial_tension_ref_temp_k', 'pour_point_max_k', 'pour_point_min_k', 'solubility', 'sulphur_fraction', 'k0y', ): assert getattr(oil_from_db, attr) == getattr(oil_from_json, attr) for db_obj, json_obj in zip(oil_from_db.cuts, oil_from_json.cuts): for attr in ('liquid_temp_k', 'vapor_temp_k', 'fraction'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.densities, oil_from_json.densities): for attr in ('kg_m_3', 'ref_temp_k', 'weathering'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.kvis, oil_from_json.kvis): for attr in ('m_2_s', 'ref_temp_k', 'weathering'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.molecular_weights, oil_from_json.molecular_weights): for attr in ('g_mol', 'ref_temp_k', 'sara_type'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.sara_densities, oil_from_json.sara_densities): for attr in ('density', 'ref_temp_k', 'sara_type'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.sara_fractions, oil_from_json.sara_fractions): for attr in ('fraction', 'ref_temp_k', 'sara_type'): assert getattr(db_obj, attr) == getattr(json_obj, attr) for db_obj, json_obj in zip(oil_from_db.categories, oil_from_json.categories): assert getattr(db_obj, 'name') == getattr(json_obj, 'name') assert (getattr(db_obj.parent, 'name') == getattr(json_obj.parent, 'name'))
''' test functions in utilities modules ''' import numpy as np import pytest from oil_library import get_oil from oil_library.utilities import (get_density, get_viscosity, get_v_max, get_boiling_points_from_api) oil_ = get_oil('LUCKENBACH FUEL OIL') # Test case - get ref temps from densities then append ref_temp for # density at 0th index for a few more values: # density_test = [d.ref_temp_k for d in oil_.densities] # density_test.append(oil_.densities[0].ref_temp_k) density_tests = [oil_.densities[ix].ref_temp_k if ix < len(oil_.densities) else oil_.densities[0].ref_temp_k for ix in range(0, len(oil_.densities) + 3)] density_exp = [d.kg_m_3 for temp in density_tests for d in oil_.densities if abs(d.ref_temp_k - temp) == 0] ''' test get_density for - scalar - list, tuple - numpy arrays as row/column and with/without output arrays '''