integer schema

barbuza · Dec 2, 2016 · b677ee9 · b677ee9
1 parent 0959769
commit b677ee9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/api/schema.py b/api/schema.py
@@ -118,6 +118,11 @@ def to_json(self):
         return {'type': 'number'}
 
 
+class Integer(Schema):
+    def to_json(self):
+        return {'type': 'integer'}
+
+
 class String(Schema):
     def to_json(self):
         return {'type': 'string'}

diff --git a/test_project/test.py b/test_project/test.py
@@ -183,14 +183,16 @@ def test_schema(self):
         schema = s.Object(
             str=s.String(),
             number=s.Optional(s.Number()),
+            integer=s.Integer(),
             null=s.Null(),
             boolean=s.Boolean(),
             array=s.Array(s.String()),
             child=child
         )
         data = {
             'str': '',
-            'number': 1,
+            'number': 1.2,
+            'integer': 2,
             'null': None,
             'boolean': True,
             'array': [''],