|
1 | 1 | import unittest
|
2 |
| -from pyvalid import accepts, ArgumentValidationError |
| 2 | +from pyvalid import accepts, ArgumentValidationError, \ |
| 3 | + InvalidArgumentNumberError |
3 | 4 | from pyvalid.validators import is_validator
|
4 | 5 |
|
5 | 6 |
|
@@ -212,6 +213,70 @@ def test_validation_func(self):
|
212 | 213 | def test_docstring(self):
|
213 | 214 | self.assertEqual(self.func_with_doc.__doc__, 'TEST_DOCSTRING')
|
214 | 215 |
|
| 216 | + def test_args_number_validation(self): |
| 217 | + # First two arguments are missing |
| 218 | + self.assertRaises(InvalidArgumentNumberError, self.func1) |
| 219 | + # The second argument is missing |
| 220 | + args = str() |
| 221 | + self.assertRaises(InvalidArgumentNumberError, self.func1, *args) |
| 222 | + # All arguments are optional there, so everything should be fine |
| 223 | + try: |
| 224 | + self.func4() |
| 225 | + except InvalidArgumentNumberError: |
| 226 | + self.fail( |
| 227 | + 'func4() unexpectedly raised the InvalidArgumentNumberError!' |
| 228 | + ) |
| 229 | + try: |
| 230 | + self.func6() |
| 231 | + except InvalidArgumentNumberError: |
| 232 | + self.fail( |
| 233 | + 'func4() unexpectedly raised the InvalidArgumentNumberError!' |
| 234 | + ) |
| 235 | + |
| 236 | + def test_ordinal(self): |
| 237 | + accepts_instance = accepts() |
| 238 | + ordinal_func = accepts_instance._Accepts__ordinal |
| 239 | + |
| 240 | + actual_result = ordinal_func(1) |
| 241 | + expected_result = '1st' |
| 242 | + self.assertEqual(actual_result, expected_result) |
| 243 | + |
| 244 | + actual_result = ordinal_func(2) |
| 245 | + expected_result = '2nd' |
| 246 | + self.assertEqual(actual_result, expected_result) |
| 247 | + |
| 248 | + actual_result = ordinal_func(3) |
| 249 | + expected_result = '3rd' |
| 250 | + self.assertEqual(actual_result, expected_result) |
| 251 | + |
| 252 | + actual_result = ordinal_func(5) |
| 253 | + expected_result = '5th' |
| 254 | + self.assertEqual(actual_result, expected_result) |
| 255 | + |
| 256 | + actual_result = ordinal_func(10) |
| 257 | + expected_result = '10th' |
| 258 | + self.assertEqual(actual_result, expected_result) |
| 259 | + |
| 260 | + actual_result = ordinal_func(12) |
| 261 | + expected_result = '12th' |
| 262 | + self.assertEqual(actual_result, expected_result) |
| 263 | + |
| 264 | + actual_result = ordinal_func(21) |
| 265 | + expected_result = '21st' |
| 266 | + self.assertEqual(actual_result, expected_result) |
| 267 | + |
| 268 | + actual_result = ordinal_func(22) |
| 269 | + expected_result = '22nd' |
| 270 | + self.assertEqual(actual_result, expected_result) |
| 271 | + |
| 272 | + actual_result = ordinal_func(23) |
| 273 | + expected_result = '23rd' |
| 274 | + self.assertEqual(actual_result, expected_result) |
| 275 | + |
| 276 | + actual_result = ordinal_func(25) |
| 277 | + expected_result = '25th' |
| 278 | + self.assertEqual(actual_result, expected_result) |
| 279 | + |
215 | 280 |
|
216 | 281 | if __name__ == '__main__':
|
217 | 282 | unittest.main()
|
0 commit comments