1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
import pytest
from pyartnet import errors
from pyartnet.base import BaseUniverse
from pyartnet.errors import ChannelNotFoundError
def test_exceptions(universe: BaseUniverse) -> None:
universe.add_channel(1, 1)
with pytest.raises(errors.ChannelExistsError) as e:
universe.add_channel(1, 1)
assert str(e.value) == 'Channel "1/1" does already exist in the universe!'
# Channels
with pytest.raises(errors.ChannelNotFoundError) as e:
universe.get_channel('2')
assert str(e.value) == 'Channel "2" not found in the universe!'
with pytest.raises(errors.ChannelNotFoundError) as e:
_ = universe['2']
assert str(e.value) == 'Channel "2" not found in the universe!'
# Overlapping channels
universe.add_channel(10, 3)
with pytest.raises(errors.OverlappingChannelError) as e:
universe.add_channel(1, 3)
assert str(e.value) == 'New channel 1/3 is overlapping with channel 1/1!'
with pytest.raises(errors.OverlappingChannelError):
universe.add_channel(9, 2)
with pytest.raises(errors.OverlappingChannelError):
universe.add_channel(12, 1)
with pytest.raises(errors.OverlappingChannelError):
universe.add_channel(8, 20)
def test_universe_resize(universe: BaseUniverse) -> None:
assert universe._data_size == 0
assert universe._data == b''
universe.add_channel(1, 1)
assert universe._data_size == 2
assert universe._data == b'\x00\x00'
universe.add_channel(6, 1)
assert universe._data_size == 6
assert universe._data == b'\x00\x00\x00\x00\x00\x00'
universe._channels.popitem()
universe.add_channel(2, 1)
assert universe._data_size == 2
assert universe._data == b'\x00\x00'
universe.add_channel(3, 1)
assert universe._data_size == 4
assert universe._data == b'\x00\x00\x00\x00'
def test_access(universe: BaseUniverse) -> None:
with pytest.raises(ChannelNotFoundError) as e:
universe.get_channel('1')
assert str(e.value) == 'Channel "1" not found in the universe!'
with pytest.raises(ChannelNotFoundError) as e:
universe.get_channel('1/1')
assert str(e.value) == 'Channel "1/1" not found in the universe!'
c = universe.add_channel(1, 1)
assert len(universe) == 1
assert universe.get_channel('1/1') is c
assert universe['1/1'] is c
c = universe.add_channel(2, 1)
assert len(universe) == 2
assert universe.get_channel('2/1') is c
assert universe['2/1'] is c
|
