def test_drag_density_setter_raises_error_for_non_number():
d = Drag(shape='sphere', diameter=10)
with pytest.raises(ValueError):
d.density = 'whoa'
def test_drag_density_setter():
d = Drag(shape='sphere', diameter=10)
d.density = 2222.2
assert 2222.2 == d.density
