-
Notifications
You must be signed in to change notification settings - Fork 0
/
tests.py
28 lines (22 loc) · 981 Bytes
/
tests.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
import unittest
import datetime
import app
class TestBartSchedule(unittest.TestCase):
def test_weekend(self):
d = datetime.datetime(year=2013, month=2, day=2, hour=8)
answer = app.bart_schedule(current_time=d)['answer']
self.assertTrue(answer == 'YES')
def test_late_night(self):
d = datetime.datetime(year=2013, month=2, day=1, hour=11)
answer = app.bart_schedule(current_time=d)['answer']
self.assertTrue(answer == 'YES')
def test_morning_commute(self):
d = datetime.datetime(year=2013, month=2, day=1, hour=8)
answer = app.bart_schedule(current_time=d)['answer']
self.assertTrue(answer == 'NO')
def test_evening_commute(self):
d = datetime.datetime(year=2013, month=2, day=1, hour=17, minute=30)
answer = app.bart_schedule(current_time=d)['answer']
self.assertTrue(answer == 'NO')
if __name__ == '__main__':
unittest.main()