1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
import pytest
from torchvision._utils import sequence_to_str
@pytest.mark.parametrize(
("seq", "separate_last", "expected"),
[
([], "", ""),
(["foo"], "", "'foo'"),
(["foo", "bar"], "", "'foo', 'bar'"),
(["foo", "bar"], "and ", "'foo' and 'bar'"),
(["foo", "bar", "baz"], "", "'foo', 'bar', 'baz'"),
(["foo", "bar", "baz"], "and ", "'foo', 'bar', and 'baz'"),
],
)
def test_sequence_to_str(seq, separate_last, expected):
assert sequence_to_str(seq, separate_last=separate_last) == expected
|
