import rpsls
import unittest
# self.assertTrue(any(x in "this sentence contains started or pending" for x in ['started', 'pending']))
# https://docs.python.org/2/library/unittest.html#assert-methods
class rpsls_test(unittest.TestCase):
def test_name_to_number(self):
self.assertEqual(rpsls.name_to_number("rock"), 0)
self.assertEqual(rpsls.name_to_number("Spock"), 1)
self.assertEqual(rpsls.name_to_number("paper"), 2)
self.assertEqual(rpsls.name_to_number("lizard"), 3)
self.assertEqual(rpsls.name_to_number("scissors"), 4)
def test_find_winner(self):
self.assertEqual(rpsls.find_winner(0, 0), 0)
self.assertEqual(rpsls.find_winner(0, 1), 2)
self.assertEqual(rpsls.find_winner(0, 2), 2)
self.assertEqual(rpsls.find_winner(0, 3), 1)
self.assertEqual(rpsls.find_winner(0, 4), 1)
self.assertEqual(rpsls.find_winner(1, 0), 1)
self.assertEqual(rpsls.find_winner(1, 1), 0)
self.assertEqual(rpsls.find_winner(1, 2), 2)
self.assertEqual(rpsls.find_winner(1, 3), 2)
self.assertEqual(rpsls.find_winner(1, 4), 1)
self.assertEqual(rpsls.find_winner(2, 0), 1)
self.assertEqual(rpsls.find_winner(2, 1), 1)
self.assertEqual(rpsls.find_winner(2, 2), 0)
self.assertEqual(rpsls.find_winner(2, 3), 2)
self.assertEqual(rpsls.find_winner(2, 4), 2)
self.assertEqual(rpsls.find_winner(3, 0), 2)
self.assertEqual(rpsls.find_winner(3, 1), 1)
self.assertEqual(rpsls.find_winner(3, 2), 1)
self.assertEqual(rpsls.find_winner(3, 3), 0)
self.assertEqual(rpsls.find_winner(3, 4), 2)
self.assertEqual(rpsls.find_winner(4, 0), 2)
self.assertEqual(rpsls.find_winner(4, 1), 2)
self.assertEqual(rpsls.find_winner(4, 2), 1)
self.assertEqual(rpsls.find_winner(4, 3), 1)
self.assertEqual(rpsls.find_winner(4, 4), 0)
if __name__ == '__main__':
unittest.main()