20 lines
351 B
Python
20 lines
351 B
Python
import unittest
|
|
from unittest.mock import patch
|
|
import mad_lib
|
|
|
|
|
|
class zyBooks_Test(unittest.TestCase):
|
|
|
|
|
|
|
|
|
|
@patch('builtins.input', return_value=['Eric', 'Chipotle', '12', 'cars'])
|
|
def test_zyBooks(self, mock_input):
|
|
result = mad_lib()
|
|
|
|
self.assertEqual(result, 6)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|