def test_print_menu_options_one_case(self, mock_input): self.assertEqual(1, crud.print_menu_options())
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)