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(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_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
'''
