github.com/phrase/openapi@v0.0.0-20240514140800-49e8a106740e/openapi-generator/templates/python/model_test.mustache (about)

     1  # coding: utf-8
     2  
     3  {{>partial_header}}
     4  
     5  from __future__ import absolute_import
     6  
     7  import unittest
     8  import datetime
     9  
    10  {{#models}}
    11  {{#model}}
    12  import {{packageName}}
    13  from {{modelPackage}}.{{classFilename}} import {{classname}}  # noqa: E501
    14  from {{packageName}}.rest import ApiException
    15  
    16  class Test{{classname}}(unittest.TestCase):
    17      """{{classname}} unit test stubs"""
    18  
    19      def setUp(self):
    20          pass
    21  
    22      def tearDown(self):
    23          pass
    24  
    25      def make_instance(self, include_optional):
    26          """Test {{classname}}
    27              include_option is a boolean, when False only required
    28              params are included, when True both required and
    29              optional params are included """
    30          # model = {{packageName}}.models.{{classFilename}}.{{classname}}()  # noqa: E501
    31  
    32          """
    33          if include_optional :
    34              return {{classname}}(
    35  {{#vars}}
    36                  {{name}} = {{#example}}{{{.}}}{{/example}}{{^example}}None{{/example}}{{^-last}}, {{/-last}}
    37  {{/vars}}
    38              )
    39          else :
    40              return {{classname}}(
    41  {{#vars}}
    42  {{#required}}
    43                  {{name}} = {{#example}}{{{.}}}{{/example}}{{^example}}None{{/example}},
    44  {{/required}}
    45  {{/vars}}
    46          )
    47          """
    48  
    49      def test{{classname}}(self):
    50          """Test {{classname}}"""
    51          inst_req_only = self.make_instance(include_optional=False)
    52          inst_req_and_optional = self.make_instance(include_optional=True)
    53  
    54  {{/model}}
    55  {{/models}}
    56  
    57  if __name__ == '__main__':
    58      unittest.main()