# coding: utf-8

"""
    Arthur GenAI Engine

    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)

    The version of the OpenAPI document: 2.1.294
    Generated by OpenAPI Generator (https://openapi-generator.tech)

    Do not edit the class manually.
"""  # noqa: E501


import unittest

from _generated.models.query_spans_response import QuerySpansResponse

class TestQuerySpansResponse(unittest.TestCase):
    """QuerySpansResponse unit test stubs"""

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def make_instance(self, include_optional) -> QuerySpansResponse:
        """Test QuerySpansResponse
            include_optional is a boolean, when False only required
            params are included, when True both required and
            optional params are included """
        # uncomment below to create an instance of `QuerySpansResponse`
        """
        model = QuerySpansResponse()
        if include_optional:
            return QuerySpansResponse(
                count = 56,
                spans = [
                    _generated.models.span_with_metrics_response.SpanWithMetricsResponse(
                        prompt_token_count = 56, 
                        completion_token_count = 56, 
                        total_token_count = 56, 
                        prompt_token_cost = 1.337, 
                        completion_token_cost = 1.337, 
                        total_token_cost = 1.337, 
                        id = '', 
                        trace_id = '', 
                        span_id = '', 
                        parent_span_id = '', 
                        span_kind = '', 
                        span_name = '', 
                        start_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                        end_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                        task_id = '', 
                        session_id = '', 
                        status_code = '', 
                        created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                        updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                        raw_data = _generated.models.raw_data.Raw Data(), 
                        input_content = '', 
                        output_content = '', 
                        metric_results = [
                            _generated.models.metric_result_response.MetricResultResponse(
                                id = '', 
                                metric_type = 'QueryRelevance', 
                                details = '', 
                                prompt_tokens = 56, 
                                completion_tokens = 56, 
                                latency_ms = 56, 
                                span_id = '', 
                                metric_id = '', 
                                created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                                updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), )
                            ], )
                    ]
            )
        else:
            return QuerySpansResponse(
                count = 56,
                spans = [
                    _generated.models.span_with_metrics_response.SpanWithMetricsResponse(
                        prompt_token_count = 56, 
                        completion_token_count = 56, 
                        total_token_count = 56, 
                        prompt_token_cost = 1.337, 
                        completion_token_cost = 1.337, 
                        total_token_cost = 1.337, 
                        id = '', 
                        trace_id = '', 
                        span_id = '', 
                        parent_span_id = '', 
                        span_kind = '', 
                        span_name = '', 
                        start_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                        end_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                        task_id = '', 
                        session_id = '', 
                        status_code = '', 
                        created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                        updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                        raw_data = _generated.models.raw_data.Raw Data(), 
                        input_content = '', 
                        output_content = '', 
                        metric_results = [
                            _generated.models.metric_result_response.MetricResultResponse(
                                id = '', 
                                metric_type = 'QueryRelevance', 
                                details = '', 
                                prompt_tokens = 56, 
                                completion_tokens = 56, 
                                latency_ms = 56, 
                                span_id = '', 
                                metric_id = '', 
                                created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                                updated_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), )
                            ], )
                    ],
        )
        """

    def testQuerySpansResponse(self):
        """Test QuerySpansResponse"""
        # inst_req_only = self.make_instance(include_optional=False)
        # inst_req_and_optional = self.make_instance(include_optional=True)

if __name__ == '__main__':
    unittest.main()
