def test_signal_dict_combine():
minibatch_size = 1
signals = SignalDict(tf.float32, minibatch_size)
key = object()
assert signals.combine([]) == []
y = signals.combine(
[
signals.get_tensor_signal([(0, 3)], key, None, (3, 2), False),
signals.get_tensor_signal([(4, 7)], key, None, (3, 2), False),
]
)
assert y.key is key
assert y._tf_indices is None
assert y.shape == (6, 2)
assert y.slices == ((0, 3), (4, 7))
y = signals.combine(
[
signals.get_tensor_signal([(0, 3)], key, None, (3, 2), False),
signals.get_tensor_signal([(3, 6)], key, None, (3, 2), False),
]
)
assert y.key is key
assert y._tf_indices is None
assert y.shape == (6, 2)
assert y.slices == ((0, 6),)
def test_signal_dict_combine():
minibatch_size = 1
signals = SignalDict(tf.float32, minibatch_size)
key = object()
assert signals.combine([]) == []
y = signals.combine(
[signals.get_tensor_signal([0, 1, 2], key, None, (3, 2), False),
signals.get_tensor_signal([4, 5, 6], key, None, (3, 2), False)])
assert y.key is key
assert y._tf_indices is None
assert y.shape == (6, 2)
assert np.all(y.indices == [0, 1, 2, 4, 5, 6])
