def test_dict_behavior():
waveform = Waveforms()
wave1 = 1.0 * np.ones(1008)
wave1_short = 1.0 * np.ones(500)
wave2 = -1.0 * np.ones(1008)
wave3 = -0.5 * np.ones(1008)
marker = 0.0 * np.ones(1008)
with pytest.raises(TypeError) as e_info:
waveform[0] = 1
with pytest.raises(RuntimeError) as e_info:
waveform[0] = (wave1, wave2, wave3, marker)
# "standart" waveform
waveform[0] = (wave1, wave2)
assert all(waveform[0][0] == wave1)
assert all(waveform[0][1] == wave2)
assert waveform[0][2] == None
assert len(waveform.get_raw_vector(0)) == 1008 * 2
# replace wave
waveform[0] = (wave1, wave3)
assert all(waveform[0][0] == wave1)
assert all(waveform[0][1] == wave3)
assert waveform[0][2] == None
# replace wave
waveform.assign_waveform(0, wave1, wave2)
assert all(waveform[0][0] == wave1)
assert all(waveform[0][1] == wave2)
assert waveform[0][2] == None
# delete wave
assert 0 in waveform.keys()
del waveform[0]
assert 0 not in waveform.keys()
# iter
waveform[0] = (wave1, wave3)
waveform[2] = (wave1, wave3)
waveform[10] = (wave1, wave3)
assert len(waveform) == 3
num_elements = 0
for _, element in waveform.items():
assert all(element[0] == wave1)
assert all(element[1] == wave3)
num_elements += 1
assert num_elements == len(waveform)
# "standart" waveform with marker
waveform[1] = (wave1, wave2, marker)
assert all(waveform[1][0] == wave1)
assert all(waveform[1][1] == wave2)
assert all(waveform[1][2] == marker)
assert len(waveform.get_raw_vector(1)) == 1008 * 3
# unequal length
with pytest.raises(RuntimeError) as e_info:
waveform[10] = (wave1_short, wave2)
with pytest.raises(RuntimeError) as e_info:
waveform[10] = (wave1, wave2, wave1_short)
# invalid inputs
with pytest.raises(RuntimeError) as e_info:
waveform[10] = (0, 0, 0)
# complex values
waveform[10] = np.ones(1008, dtype=np.complex128)
waveform[10] = (np.ones(1008, dtype=np.complex128), marker)
waveform[10] = (np.ones(1008, dtype=np.complex128), None, marker)
with pytest.raises(RuntimeError) as e_info:
waveform[10] = (np.ones(1008, dtype=np.complex128), wave2, None)
def test_sequence_snippet():
waveform = Waveforms()
waveform[0] = (Wave(np.ones(1008), "w1", 3), Wave(np.ones(1008), "w2", 2))
waveform[3] = (Wave(np.ones(252), "w3"), np.ones(252), 1 * np.ones(252))
waveform[1] = (np.ones(504), np.ones(504), 15 * np.ones(504))
waveform[5] = (np.ones(252),)
result = waveform.get_sequence_snippet()
assert (
result
== """\
wave w1 = placeholder(1008, false, false);
wave w2 = placeholder(1008, false, false);
assignWaveIndex(1, 2, w1, 2, w2, 0);
assignWaveIndex(placeholder(504, true, true), placeholder(504, true, true), 1);
wave w3 = placeholder(252, true, false);
assignWaveIndex(w3, placeholder(252, false, false), 3);
assignWaveIndex(placeholder(252, false, false), 5);"""
)
waveform = Waveforms()
waveform.assign_waveform(
0,
Wave(np.ones(1008), name="w1", output=OutputType.OUT1 | OutputType.OUT2),
Wave(np.ones(1008), name="w2", output=OutputType.OUT1 | OutputType.OUT2),
np.ones(1008),
)
waveform.assign_waveform(
3,
np.ones(1008),
np.ones(1008),
np.ones(1008),
)
waveform.assign_waveform(
5,
np.ones(1008),
np.ones(1008),
)
waveform.assign_waveform(
4,
Wave(np.ones(1008), name="test1", output=OutputType.OUT2),
Wave(np.ones(1008), name="test2", output=OutputType.OUT1),
)
waveform[2] = (np.ones(1008, dtype=np.complex128), None, 15 * np.ones(1008))
waveform[1] = (
Wave(
np.ones(1008, dtype=np.complex128),
name=["comp1", "comp2"],
output=[OutputType.OUT2, OutputType.OUT1 | OutputType.OUT2],
),
)
result = waveform.get_sequence_snippet()
assert (
result
== """\
wave w1 = placeholder(1008, true, false);
wave w2 = placeholder(1008, false, false);
assignWaveIndex(1, 2, w1, 1, 2, w2, 0);
wave comp1 = placeholder(1008, false, false);
wave comp2 = placeholder(1008, false, false);
assignWaveIndex(2, comp1, 1, 2, comp2, 1);
assignWaveIndex(placeholder(1008, true, true), placeholder(1008, true, true), 2);
assignWaveIndex(placeholder(1008, true, false), placeholder(1008, false, false), 3);
wave test1 = placeholder(1008, false, false);
wave test2 = placeholder(1008, false, false);
assignWaveIndex(2, test1, 1, test2, 4);
assignWaveIndex(placeholder(1008, false, false), placeholder(1008, false, false), 5);"""
)