def test_timeout_set_handle_error(self): """Test handling an error in setting the timeout""" shell = VisaShell() shell.current = True temp_stdout = StringIO() with redirect_stdout(temp_stdout): shell.do_timeout("1000") output = temp_stdout.getvalue() assert "no attribute" in output
def test_timeout_get_handle_error(self): """Test handling an error in getting teh timeout. """ shell = VisaShell() shell.current = True temp_stdout = StringIO() with redirect_stdout(temp_stdout): shell.do_timeout("") output = temp_stdout.getvalue() self.assertIn('no attribute', output)