from citygrid import PrivacyBudget
def test_privacy_budget_basic():
pb = PrivacyBudget(signal="der_voltage", budget=0.5, expiry=None)
assert pb.budget == 0.5
assert pb.signal == "der_voltage"