Python flatten示例

示例#1

def env():
    net = Network.load('net/net_test.yaml')
    env = gym.make('openmodelica_microgrid_gym:ModelicaEnv_test-v1',
                   viz_mode=None,
                   model_path='omg_grid/test.fmu',
                   max_episode_steps=100,
                   net=net)

    return env, net.in_vars(), flatten(net.out_vars())

示例#2

文件： test_runner.py 项目： zhangtjtongxue/openmodelica-microgrid-gym

def env():
    model_input = ['i1p1', 'i1p2', 'i1p3', 'i2p1', 'i2p2', 'i2p3']
    conf = {'lc1': [['inductor1.i', 'inductor2.i', 'inductor3.i'],
                    ['capacitor1.v', 'capacitor2.v', 'capacitor3.v']],
            'lcl1': [['inductor1.i', 'inductor2.i', 'inductor3.i'],
                     ['capacitor1.v', 'capacitor2.v', 'capacitor3.v']]}
    env = gym.make('openmodelica_microgrid_gym:ModelicaEnv_test-v1',
                   viz_mode=None,
                   model_path='fmu/test.fmu',
                   max_episode_steps=100,
                   model_input=model_input,
                   model_output=conf)

    return env, model_input, flatten(conf)

示例#3

def test_flatten(i, o):
    assert flatten(*i) == o