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"