def setUp(self):
self.c = ComboLock(5, 6, 7)
class ComboLockTests(unittest.TestCase):
def setUp(self):
self.c = ComboLock(5, 6, 7)
def test_assign_secret(self):
self.assertEqual(5, self.c.assign_secret(5))
def test_assign_secret_below_0(self):
self.assertEqual(0, self.c.assign_secret(-5))
def test_assign_secret_beyond_39(self):
self.assertEqual(39, self.c.assign_secret(70))
def test_get_dial(self):
self.assertEqual(0, self.c.get_dial())
def test_reset(self):
self.c.reset()
self.assertEqual(0, self.c.get_dial())
def test_dial_scrolling(self):
self.assertEqual(3, self.c.dial_scrolling(3))
def test_dial_scrolling_beyond_39(self):
self.assertEqual(1, self.c.dial_scrolling(40))
def test_dial_scrolling_beyond_39_again(self):
self.assertEqual(31, self.c.dial_scrolling(70))
def test_dial_scrolling_below_0(self):
self.assertEqual(39, self.c.dial_scrolling(-1))
def test_turn_right(self):
self.c.turn_right(5)
self.assertEqual(5, self.c.get_dial())
def test_turn_left(self):
self.c.turn_left(5)
self.assertEqual(40-5, self.c.get_dial())
def test_dial_to_39_and_move_right(self):
self.c._dial = 39
self.c.turn_right(1)
self.assertEqual(0, self.c.get_dial())
def test_dial_to_0_and_move_left(self):
self.c._dial = 0
self.c.turn_left(1)
self.assertEqual(39, self.c.get_dial())
