-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_day13.py
47 lines (37 loc) · 1.16 KB
/
test_day13.py
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
import unittest
import aoc
import day13
class TestDay13( unittest.TestCase ):
example_input_part1 = [
r'/->-\ ',
r'| | /----\\', # Ugh. Need to add escape to slash so that we don't have \'.
r'| /-+--+-\ |',
r'| | | | v |',
r'\-+-/ \-+--/',
r' \------/ '
]
example_input_part2 = [
r'/>-<\ ',
r'| | ',
r'| /<+-\\',
r'| | | v',
r'\>+</ |',
r' | ^',
r' \<->/'
]
#@unittest.skip( 'dev' )
def test_part1_example1( self ):
self.assertEqual( day13.part1( self.example_input_part1 ), '7,3' )
#@unittest.skip( 'dev' )
def test_part1_input( self ):
result = day13.part1( aoc.read_input( 'day13.input' ) )
self.assertEqual( result, '38,57' )
#@unittest.skip( 'dev' )
def test_part2_example1( self ):
self.assertEqual( day13.part2( self.example_input_part2 ), '6,4' )
#@unittest.skip( 'dev' )
def test_part2_input( self ):
result = day13.part2( aoc.read_input( 'day13.input' ) )
self.assertEqual( result, '4,92' )
if __name__ == '__main__':
unittest.main()