Add Python 3.10 to Python version (#765)

This also add simple test to ensure that All the supported versions of Python are listed. Co-authored-by: Samuel Gaist <samuel,gaist@idiap.ch>
koxudaxi · May 16, 2022 · 6e38bf7 · 6e38bf7
1 parent 560c45b
commit 6e38bf7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/datamodel_code_generator/format.py b/datamodel_code_generator/format.py
@@ -13,6 +13,7 @@ class PythonVersion(Enum):
     PY_37 = '3.7'
     PY_38 = '3.8'
     PY_39 = '3.9'
+    PY_310 = '3.10'
 
     @property
     def has_literal_type(self) -> bool:

diff --git a/tests/test_format.py b/tests/test_format.py
@@ -0,0 +1,9 @@
+import sys
+
+from datamodel_code_generator.format import PythonVersion
+
+
+def test_python_version():
+    """Ensure that the python version used for the tests is properly listed"""
+
+    _ = PythonVersion("{}.{}".format(*sys.version_info[:2]))