def test_two_equal_size_fuelbeds(self):
g = {
"fuelbeds": [{
"fccs_id": "46",
"pct": 100.0
}, {
"fccs_id": "44",
"pct": 100.0
}]
}
assert "46" == firescsvs._pick_representative_fuelbed({}, g)
def test_three_fuelbed(self):
g = {
"fuelbeds": [{
"fccs_id": "46",
"pct": 10.0
}, {
"fccs_id": "47",
"pct": 60.0
}, {
"fccs_id": "48",
"pct": 30.0
}]
}
assert "47" == firescsvs._pick_representative_fuelbed({}, g)
def test_invalid_fuelbed(self):
g = {"fuelbeds": 'sdf'}
with raises(TypeError) as e_info:
firescsvs._pick_representative_fuelbed({}, g)
# TODO: assert e_info.value.args[0] == '...''
g = {"fuelbeds": [{"sdf_fccs_id": "46", "pct": 100.0}]}
with raises(KeyError) as e_info:
firescsvs._pick_representative_fuelbed({}, g)
# TODO: assert e_info.value.args[0] == '...''
g = {"fuelbeds": [{"fccs_id": "46", "sdfsdfpct": 100.0}]}
with raises(KeyError) as e_info:
firescsvs._pick_representative_fuelbed({}, g)
def test_one_fuelbed(self):
g = {"fuelbeds": [{"fccs_id": "46", "pct": 100.0}]}
assert "46" == firescsvs._pick_representative_fuelbed({}, g)
def test_no_fuelbeds(self):
g = {}
assert None == firescsvs._pick_representative_fuelbed({}, g)
g = {"activity": []}
assert None == firescsvs._pick_representative_fuelbed({}, g)
def test_two_fuelbeds_one_with_no_pct(self):
g = {"fuelbeds": [{"fccs_id": "46"}, {"fccs_id": "44", "pct": 100.0}]}
assert "44" == firescsvs._pick_representative_fuelbed({}, g)
def test_one_fuelbed_no_pct(self):
g = {"fuelbeds": [{"fccs_id": "46"}]}
assert "46" == firescsvs._pick_representative_fuelbed({}, g)
def test_one_fuelbed_no_fccs_id(self):
g = {"fuelbeds": [{"pct": 100.0}]}
assert None == firescsvs._pick_representative_fuelbed({}, g)
g["fuelbeds"][0]["sdf_fccs_id"] = "46"
assert None == firescsvs._pick_representative_fuelbed({}, g)
def test_invalid_fuelbed(self):
g = {"fuelbeds": 'sdf'}
with raises(AttributeError) as e_info:
firescsvs._pick_representative_fuelbed({}, g)