def test_max_value_int_smaller(self): with mock.patch("yehua.thirdparty.cutie.input", return_value="1"): self.assertEqual(cutie.get_number("foo", max_value=2), 1)
def test_max_value_int_too_high(self, mock_print): with mock.patch("yehua.thirdparty.cutie.input", return_value="2"): with self.assertRaises(MockException): cutie.get_number("foo", max_value=1) mock_print.assert_called_once_with( "Has to be at most 1.\033[1A\r\033[K", end="")
def test_invalid_number(self, mock_print): with mock.patch("yehua.thirdparty.cutie.input", return_value="foo"): with self.assertRaises(MockException): cutie.get_number("bar") mock_print.assert_called_once_with( "Not a valid number.\033[K\033[1A\r\033[K", end="")
def test_min_value_int_greater(self): with mock.patch("yehua.thirdparty.cutie.input", return_value="2"): self.assertEqual(cutie.get_number("foo", min_value=1), 2)
def test_max_value_float_equal(self): with mock.patch("yehua.thirdparty.cutie.input", return_value="1.1"): self.assertEqual(cutie.get_number("foo", max_value=1.1), 1.1)
def test_min_value_int_too_low(self, mock_print): with mock.patch("yehua.thirdparty.cutie.input", return_value="1"): with self.assertRaises(MockException): cutie.get_number("foo", min_value=2) mock_print.assert_called_once_with( "Has to be at least 2.\033[K\033[1A\r\033[K", end="")
def test_not_allow_float_returns_int(self): with mock.patch("yehua.thirdparty.cutie.input", return_value="1"): val = cutie.get_number("foo", allow_float=False) self.assertIsInstance(val, int) self.assertEqual(val, 1)
def test_allow_float_returns_float(self): with mock.patch("yehua.thirdparty.cutie.input", return_value="1.2"): val = cutie.get_number("foo") self.assertIsInstance(val, float) self.assertEqual(val, 1.2)
def test_not_allow_float(self, mock_print): with mock.patch("yehua.thirdparty.cutie.input", return_value="1.2"): with self.assertRaises(MockException): cutie.get_number("foo", allow_float=False) mock_print.assert_called_once_with( "Has to be an integer.\033[K\033[1A\r\033[K", end="")
def test_print_finalize(self, mock_print): with mock.patch("yehua.thirdparty.cutie.input", return_value="1"): cutie.get_number("foo") mock_print.assert_called_once_with("\033[K", end="")