def test_raises_when_spec_in_kwargs(self) -> None:
with pytest.raises(ValueError) as e:
bps.double_stack(['a', 'b'], 'foo', 'bar', foo=10)
assert str(e.value).endswith("Stack property 'foo' cannot appear in keyword args")
with pytest.raises(ValueError) as e:
bps.double_stack(['a', 'b'], 'foo', 'bar', bar=10)
assert str(e.value).endswith("Stack property 'bar' cannot appear in keyword args")
def test_raises_when_kwargs_list_lengths_and_stackers_lengths_differ(self):
with pytest.raises(ValueError) as e:
bps.double_stack(['a', 'b', 'c'],
'foo',
'bar',
baz=[1, 2],
quux=[3, 4])
assert str(e.value).endswith(
"Keyword argument sequences for broadcasting must be the same length as stackers"
)
def test_raises_when_kwargs_list_lengths_differ(self) -> None:
with pytest.raises(ValueError) as e:
bps.double_stack(['a', 'b'],
'foo',
'bar',
baz=[1, 2],
quux=[3, 4, 5])
assert str(e.value).endswith(
"Keyword argument sequences for broadcasting must all be the same lengths. Got lengths: [2, 3]"
)
def test_broadcast_with_no_kwargs(self) -> None:
stackers = ['a', 'b', 'c', 'd']
kws = bps.double_stack(stackers, 'start', 'end')
assert len(kws) == len(stackers)
for i, kw in enumerate(kws):
assert {"start", "end", "name"} == set(kw.keys())
assert list(kw["start"]["expr"].fields) == stackers[:i]
assert list(kw["end"]["expr"].fields) == stackers[: (i + 1)]
def test_broadcast_with_no_kwargs(self):
stackers = ['a', 'b', 'c', 'd']
kws = bps.double_stack(stackers, 'start', 'end')
assert len(kws) == len(stackers)
for i, kw in enumerate(kws):
assert set(['start', 'end', 'name']) == set(kw.keys())
assert list(kw['start']['expr'].fields) == stackers[:i]
assert list(kw['end']['expr'].fields) == stackers[:(i + 1)]
def test_broadcast_name_scalar_overrides(self) -> None:
stackers = ['a', 'b', 'c', 'd']
kws = bps.double_stack(stackers, 'start', 'end', foo=[10, 20, 30, 40], bar="baz", name="name")
assert len(kws) == len(stackers)
for i, kw in enumerate(kws):
assert {"start", "end", "foo", "bar", "name"} == set(kw.keys())
assert list(kw["start"]["expr"].fields) == stackers[:i]
assert list(kw["end"]["expr"].fields) == stackers[: (i + 1)]
assert kw["foo"] == [10, 20, 30, 40][i]
assert kw["bar"] == "baz"
assert kw["name"] == "name"
def test_broadcast_with_scalar_kwargs(self):
stackers = ['a', 'b', 'c', 'd']
kws = bps.double_stack(stackers, 'start', 'end', foo=10, bar="baz")
assert len(kws) == len(stackers)
for i, kw in enumerate(kws):
assert set(['start', 'end', 'foo', 'bar',
'name']) == set(kw.keys())
assert list(kw['start']['expr'].fields) == stackers[:i]
assert list(kw['end']['expr'].fields) == stackers[:(i + 1)]
assert kw['foo'] == 10
assert kw['bar'] == "baz"
assert kw['name'] == stackers[i]
def test_broadcast_name_scalar_overrides(self):
stackers = ['a', 'b', 'c', 'd']
kws = bps.double_stack(stackers,
'start',
'end',
foo=[10, 20, 30, 40],
bar="baz",
name="name")
assert len(kws) == len(stackers)
for i, kw in enumerate(kws):
assert set(['start', 'end', 'foo', 'bar',
'name']) == set(kw.keys())
assert list(kw['start']['expr'].fields) == stackers[:i]
assert list(kw['end']['expr'].fields) == stackers[:(i + 1)]
assert kw['foo'] == [10, 20, 30, 40][i]
assert kw['bar'] == "baz"
assert kw['name'] == "name"
