def test_verify_unit():
# Implicit dimensionless values are allowed
verify_unit(0, u.one)
# Test failure mode
with pytest.raises(ValueError):
verify_unit(0, u.meter)
# Quantity should be passed back if unit matches
assert verify_unit(1 * u.meter, u.meter) == 1 * u.meter
def test_verify_unit():
# Implicit dimensionless values are allowed, test that Quantity is returned.
assert verify_unit(0, u.one) == 0 * u.one
assert verify_unit(0, '') == 0 * u.one
# Test failure mode
with pytest.raises(ValueError):
verify_unit(0, u.meter)
with pytest.raises(ValueError):
verify_unit(0, 'm')
# Quantity should be passed back if unit matches
assert verify_unit(1 * u.meter, u.meter) == 1 * u.meter
assert verify_unit(1 * u.meter, 'm') == 1 * u.meter
