test_generator.py 文件源码

def test_default_generator(default_generator):

    g = default_generator
    assert g.name == 'GenCo1'
    assert g.generator_bus == 'Bus1'
    assert g.generation_type == 'NATURALGAS'

    assert g.maximum_real_power == 100

    assert g.ramp_up_rate == 100
    assert g.ramp_down_rate == 100

    assert len(g.cost_curve_points) == 4
    assert len(g.cost_curve_values) == 4

    with pt.raises(T.TraitError):
        g.ramp_up_rate = 100.5

    with pt.raises(T.TraitError):
        g.ramp_down_rate = 100.5

    assert g.ramp_up_rate == 100
    assert g.ramp_down_rate == 100

    with pt.raises(AttributeError) as excinfo:
        g.ramp_rate

    assert 'ramp_down_rate' in str(excinfo.value) and 'ramp_up_rate' in str(excinfo.value)

    with pt.raises(T.TraitError):
        g.nsegments = 0

    with pt.raises(T.TraitError):
        g.initial_real_power = 100.5

    with pt.raises(T.TraitError):
        g.initial_imag_power = 100.5

    with pt.raises(T.TraitError) as excinfo:
        g.cost_curve_points = [0, 1, 2]

    assert 'must be equal to' in str(excinfo.value)

    with pt.raises(T.TraitError) as excinfo:
        g.cost_curve_values = [0, 1, 2]

    assert 'must be equal to' in str(excinfo.value)