def test_get_install_arg(self, mock_input): mock_input.return_value = 'develop' i = Setup() arg = i.get_install_arg() self.assertEqual(arg, 'develop')
def test_get_install_arg_input_error(self, mock_input): mock_input.return_value = 'foo' i = Setup() arg = i.get_install_arg() e = self.mocked_facio_hooks_python_setup_Setup_error self.assertEqual(arg, None) e.assert_called_with("You did not enter a valid setup.py arg")