File size: 1,400 Bytes
e69d4e4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | """Unit tests for the unit_lookup function."""
import copd
import pandas as pd
def test_unit_lookup_defined():
"""Test unit lookup output for all defined unit codes."""
assert copd.unit_lookup(pd.Series(0)) == 'Count'
assert copd.unit_lookup(pd.Series(1)) == 'CountPerSecond'
assert copd.unit_lookup(pd.Series(2)) == 'InternationalUnit'
assert copd.unit_lookup(pd.Series(3)) == 'Joule'
assert copd.unit_lookup(pd.Series(4)) == 'Kelvin'
assert copd.unit_lookup(pd.Series(5)) == 'Kilogram'
assert copd.unit_lookup(pd.Series(6)) == 'KilogramPerLiter'
assert copd.unit_lookup(pd.Series(7)) == 'KilogramPerSquareMeter'
assert copd.unit_lookup(pd.Series(8)) == 'Liter'
assert copd.unit_lookup(pd.Series(9)) == 'LiterPerKilogramSecond'
assert copd.unit_lookup(pd.Series(10)) == 'LiterPerSecond'
assert copd.unit_lookup(pd.Series(11)) == 'Meter'
assert copd.unit_lookup(pd.Series(12)) == 'Pascal'
assert copd.unit_lookup(pd.Series(13)) == 'Percent'
assert copd.unit_lookup(pd.Series(14)) == 'Second'
assert copd.unit_lookup(pd.Series(15)) == 'Siemen'
def test_unit_lookup_undefined():
"""Test unit lookup output for undefined unit code and other input."""
assert copd.unit_lookup(pd.Series(16)) == 'Undefined'
assert copd.unit_lookup(pd.Series(42)) == 'Undefined'
assert copd.unit_lookup(pd.Series('A')) == 'Undefined'
|