typed_input/float_input_test.py at main · sashsinha/typed_input · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import io
import unittest
from unittest import mock

from typed_input import float_input


class FloatInputTest(unittest.TestCase):
  @mock.patch('builtins.input', return_value='42.5')
  @mock.patch('sys.stdout', new_callable=io.StringIO)
  def test_valid_input_no_constraints(self, mock_stdout, mock_input):
    self.assertEqual(float_input(), 42.5)
    mock_input.assert_called_once_with()
    self.assertEqual(mock_stdout.getvalue().strip(), '')

  @mock.patch('builtins.input', return_value='42.5')
  @mock.patch('sys.stdout', new_callable=io.StringIO)
  def test_valid_input_with_prompt(self, mock_stdout, mock_input):
    self.assertEqual(float_input(prompt='Enter a float: '), 42.5)
    mock_input.assert_called_once_with('Enter a float: ')
    self.assertEqual(mock_stdout.getvalue().strip(), '')

  @mock.patch('builtins.input', side_effect=['abc', '42.5'])
  @mock.patch('sys.stdout', new_callable=io.StringIO)
  def test_invalid_input_retry(self, mock_stdout, mock_input):
    self.assertEqual(float_input(), 42.5)
    self.assertEqual(mock_input.call_count, 2)
    self.assertIn(
      'Error: You must enter a valid float.', mock_stdout.getvalue()
    )

  @mock.patch('builtins.input', side_effect=['3.0', '5.0'])
  @mock.patch('sys.stdout', new_callable=io.StringIO)
  def test_min_value_violation(self, mock_stdout, mock_input):
    self.assertEqual(float_input(min_value=5.0), 5.0)
    self.assertEqual(mock_input.call_count, 2)
    self.assertIn('Error: Value must be at least 5.0.', mock_stdout.getvalue())

  @mock.patch('builtins.input', side_effect=['10.0', '8.0'])
  @mock.patch('sys.stdout', new_callable=io.StringIO)
  def test_max_value_violation(self, mock_stdout, mock_input):
    self.assertEqual(float_input(max_value=8.0), 8.0)
    self.assertEqual(mock_input.call_count, 2)
    self.assertIn('Error: Value must be at most 8.0.', mock_stdout.getvalue())

  @mock.patch('builtins.input', side_effect=['2.0', '12.0', '8.0'])
  @mock.patch('sys.stdout', new_callable=io.StringIO)
  def test_min_and_max_value_violation(self, mock_stdout, mock_input):
    self.assertEqual(float_input(min_value=5.0, max_value=10.0), 8.0)
    self.assertEqual(mock_input.call_count, 3)
    self.assertIn('Error: Value must be at least 5.0.', mock_stdout.getvalue())
    self.assertIn('Error: Value must be at most 10.0.', mock_stdout.getvalue())

  @mock.patch('builtins.input', side_effect=['abc', '3.0', '5.0'])
  @mock.patch('sys.stdout', new_callable=io.StringIO)
  def test_invalid_input_then_out_of_range(self, mock_stdout, mock_input):
    self.assertEqual(float_input(min_value=5.0), 5.0)
    self.assertEqual(mock_input.call_count, 3)
    self.assertIn(
      'Error: You must enter a valid float.', mock_stdout.getvalue()
    )
    self.assertIn('Error: Value must be at least 5.0.', mock_stdout.getvalue())

  @mock.patch('builtins.input', side_effect=[''])
  @mock.patch('sys.stdout', new_callable=io.StringIO)
  def test_default_value_used_when_input_empty(
    self, usued_mock_stdout, mock_input
  ):
    self.assertEqual(float_input(default_value=10.5), 10.5)
    mock_input.assert_called_once_with()

  @mock.patch('builtins.input', side_effect=['abc', '42.5'])
  @mock.patch('sys.stdout', new_callable=io.StringIO)
  def test_custom_type_error_message(self, mock_stdout, usued_mock_input):
    self.assertEqual(
      float_input(type_error_message='Must provide a valid height.'), 42.5
    )
    self.assertIn('Must provide a valid height.', mock_stdout.getvalue())

  def test_error_raised_when_min_value_greater_than_max_value(self):
    with self.assertRaises(ValueError) as context:
      float_input(min_value=2.0, max_value=1.0)
    self.assertEqual(
      str(context.exception), '(min_value=2.0) is greater than (max_value=1.0).'
    )

  def test_error_raised_when_default_value_less_than_min_value(self):
    with self.assertRaises(ValueError) as context:
      float_input(min_value=2.0, default_value=1.0)
    self.assertEqual(
      str(context.exception),
      '(default_value=1.0) is less than (min_value=2.0).',
    )

  def test_error_raised_when_default_value_greater_than_max_value(self):
    with self.assertRaises(ValueError) as context:
      float_input(max_value=1.0, default_value=2.0)
    self.assertEqual(
      str(context.exception),
      '(default_value=2.0) is greater than (max_value=1.0).',
    )