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
|
from cigar import Cigar
def test_basic():
#Test 100M
c = Cigar('100M')
assert len(c) == 100
assert list(c.items()) == [(100, 'M')]
# Test '20H20M20S'
c = Cigar('20H20M20S')
assert len(c) == 40
assert str(c) == '20H20M20S'
assert list(c.items()) == [(20, 'H'), (20, 'M'), (20, 'S')]
assert c.mask_left(29).cigar, c.cigar == ('20H9S11M20S', '20H20M20S')
# Test '10M20S10M'
c = Cigar('10M20S10M')
assert c.mask_left(10).cigar == '30S10M'
assert c.mask_left(9).cigar == '9S1M20S10M'
#Custom Tests
assert [Cigar('10S').mask_left(10).cigar, Cigar('10H').mask_left(10).cigar, Cigar('10H').mask_left(11).cigar, Cigar('10H').mask_left(9).cigar] == ['10S', '10H', '10H', '10H']
assert Cigar('1M10H').mask_left(9).cigar, Cigar('5M10H').mask_left(9).cigar == ('1S10H', '5S10H')
c = Cigar('1S1H1S5H1S5M10H')
assert c.mask_left(9).cigar == c.cigar
c = Cigar('1S1H1S5H1S5M10H')
assert c.mask_right(9).cigar == c.cigar
assert c.mask_right(11).cigar == '1S1H1S5H1S4M1S10H'
|
