def test_set_unit(unit): f = Figure() assert None == f.unit f.unit = unit assert unit == f.unit f.unit = None assert None == f.unit f.set_unit(unit) assert unit == f.unit