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()