def test_set_theta(): v = Vector(3, 4) v.theta = 0.0 assert math.isclose(v.x, 5.0) assert math.isclose(v.y, 0.0) v.theta = math.pi / 2.0 assert math.isclose(v.x, 0.0) assert math.isclose(v.y, 5.0)