feat(FIS-8339): Add NUMBER to accept both int and float

CHANGELOG.md

@@ -1,6 +1,11 @@
 Changelog
 =========
 
+3.5.0
+-----
+- [feature] add `NUMBER` that accepts float and integer
+
+
 3.4.1
 -----
 - Fix `dsv` decorator to work with `DSVError`.

README.md

@@ -103,6 +103,9 @@ validate_data_spec(multirow_data, SingleSpec, multirow=True) # return True
 ### FLOAT
 `float_field = Checker([FLOAT])` or `Checker([float])`
 
+### NUMBER
+`number_field = Checker([NUMBER])`
+
 ### STR
 `str_field = Checker([STR])` or `Checker([str])`
 

data_spec_validator/__version__.py

@@ -1 +1 @@
-__version__ = '3.4.1'
+__version__ = '3.5.0'

data_spec_validator/spec/__init__.py

@@ -25,6 +25,7 @@
     LIST,
     LIST_OF,
     NONE,
+    NUMBER,
     ONE_OF,
     REGEX,
     SELF,
@@ -66,6 +67,7 @@
     "LIST",
     "LIST_OF",
     "NONE",
+    "NUMBER",
     "ONE_OF",
     "REGEX",
     "SELF",

data_spec_validator/spec/checks.py

@@ -28,6 +28,7 @@
     LIST,
     LIST_OF,
     NONE,
+    NUMBER,
     ONE_OF,
     RAW_CHECK_TYPE,
     REGEX,
@@ -68,6 +69,7 @@ def _get_default_check_2_validator_map() -> Dict[str, BaseValidator]:
         ListOfValidator,
         ListValidator,
         NoneValidator,
+        NumberValidator,
         OneOfValidator,
         RegexValidator,
         SpecValidator,
@@ -79,6 +81,7 @@ def _get_default_check_2_validator_map() -> Dict[str, BaseValidator]:
     return {
         INT: IntValidator(),
         FLOAT: FloatValidator(),
+        NUMBER: NumberValidator(),
         STR: StrValidator(),
         DIGIT_STR: DigitStrValidator(),
         BOOL: BoolValidator(),

data_spec_validator/spec/defines.py

@@ -6,6 +6,7 @@
 NONE = 'none'
 INT = 'int'
 FLOAT = 'float'
+NUMBER = 'number'
 DIGIT_STR = 'digit_str'  # URL params cannot distinguish from strings and numbers
 STR = 'str'
 BOOL = 'bool'

data_spec_validator/spec/validators.py

@@ -34,6 +34,7 @@
     LIST,
     LIST_OF,
     NONE,
+    NUMBER,
     ONE_OF,
     REGEX,
     SPEC,
@@ -216,6 +217,16 @@ def validate(value, extra, data):
         return ok, info
 
 
+class NumberValidator(BaseValidator):
+    name = NUMBER
+
+    @staticmethod
+    def validate(value, extra, data):
+        ok = type(value) is float or type(value) is int
+        info = '' if ok else TypeError(f'{repr(value)} is not a number')
+        return ok, info
+
+
 class StrValidator(BaseValidator):
     name = STR
 

test/test_spec.py

@@ -26,6 +26,7 @@
     LIST,
     LIST_OF,
     NONE,
+    NUMBER,
     ONE_OF,
     REGEX,
     SELF,
@@ -67,6 +68,19 @@ class FloatSpec:
         nok_data = dict(float_field=3)
         assert is_something_error(TypeError, validate_data_spec, nok_data, FloatSpec)
 
+    def test_number(self):
+        class NumberSpec:
+            number_field = Checker([NUMBER])
+
+        ok_data = dict(number_field=3.0)
+        assert validate_data_spec(ok_data, NumberSpec)
+
+        ok_data = dict(number_field=3)
+        assert validate_data_spec(ok_data, NumberSpec)
+
+        nok_data = dict(number_field='3.0')
+        assert is_something_error(TypeError, validate_data_spec, nok_data, NumberSpec)
+
     def test_str(self):
         class StrSpec:
             str_field = Checker([STR])