예제 #1
0
 def test_print_menu_options_one_case(self, mock_input):
     self.assertEqual(1, crud.print_menu_options())
예제 #2
0
 def test_print_menu_options_printed_output(self, mock_stdout, mock_input):
     crud.print_menu_options()
     expected = (
         ' 1.Add Student \n 2.Delete Student \n 3.Calculate Class Average \n'
         ' 4.Print Class List \n 5.Add Grade To Student Record \n 6.Quit\n')
     self.assertEqual(mock_stdout.getvalue(), expected)