예제 #1
0
    def test_build_from_config2(self):
        # Test rebuilding a sequence including a template sequence.
        # Channel mapping of template_vars values are known.
        conf = {'template_id': 'test', 'name': 'Template',
                'template_vars': "{'b': '25'}"}
        seq = TemplateSequence.build_from_config(conf, self.dependecies)
        seq.context.channel_mapping = {'A': 'Ch1_L', 'B': 'Ch2_L',
                                       'Ch1': 'Ch2_A', 'Ch2': 'Ch1_A'}
        root = RootSequence()
        context = TestContext(sampling=0.5)
        root.context = context
        root.items = [seq]
        pref = root.preferences_from_members()

        new = RootSequence.build_from_config(pref, self.dependecies)
        assert_equal(new.items[0].index, 1)

        seq = new.items[0]
        assert_equal(seq.name, 'Template')
        assert_equal(seq.template_id, 'test')
        assert_equal(seq.template_vars, dict(b='25'))
        assert_equal(seq.local_vars, dict(a='1.5'))
        assert_equal(len(seq.items), 4)
        assert_equal(seq.items[3].index, 5)
        assert_equal(seq.docs, 'Basic user comment\nff')

        context = seq.context
        assert_equal(context.template, seq)
        assert_equal(context.logical_channels, ['A', 'B'])
        assert_equal(context.analogical_channels, ['Ch1', 'Ch2'])
        assert_equal(context.channel_mapping, {'A': 'Ch1_L', 'B': 'Ch2_L',
                                               'Ch1': 'Ch2_A', 'Ch2': 'Ch1_A'})
예제 #2
0
def test_build_from_config():
    # Test building a pulse sequence.
    root = RootSequence()
    context = BaseContext()
    root.context = context
    root.external_vars = {'a': 1.5}

    pulse1 = Pulse(def_1='1.0', def_2='{a}')
    pulse2 = Pulse(def_1='{a} + 1.0', def_2='3.0')
    pulse3 = Pulse(def_1='{2_stop} + 0.5', def_2='10',
                   kind='Analogical', shape=SquareShape())
    seq = Sequence(items=[Pulse(def_1='{2_stop} + 0.5', def_2='10',
                                kind='Analogical', shape=SquareShape())])
    root.items.extend([pulse1, pulse2, pulse3, seq])

    pref = root.preferences_from_members()
    dependecies = {'pulses': {'Sequence': Sequence, 'Pulse': Pulse,
                              'shapes': {'SquareShape': SquareShape},
                              'contexts': {'BaseContext': BaseContext}}}

    aux = RootSequence.build_from_config(pref, dependecies)
    assert_equal(aux.external_vars, {'a': 1.5})
    assert_equal(len(aux.items), 4)
    assert_is_instance(aux.context, BaseContext)

    pulse1 = aux.items[0]
    assert_equal(pulse1.def_1, '1.0')
    assert_equal(pulse1.def_2, '{a}')

    pulse2 = aux.items[1]
    assert_equal(pulse2.def_1, '{a} + 1.0')
    assert_equal(pulse2.def_2, '3.0')

    pulse3 = aux.items[2]
    assert_equal(pulse3.def_1, '{2_stop} + 0.5')
    assert_equal(pulse3.def_2, '10')
    assert_equal(pulse3.kind, 'Analogical')
    assert_is_instance(pulse3.shape, SquareShape)

    seq = aux.items[3]
    assert_equal(len(seq.items), 1)