# 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.trace_response import TraceResponse

class TestTraceResponse(unittest.TestCase):
    """TraceResponse unit test stubs"""

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def make_instance(self, include_optional) -> TraceResponse:
        """Test TraceResponse
            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 `TraceResponse`
        """
        model = TraceResponse()
        if include_optional:
            return TraceResponse(
                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,
                trace_id = '',
                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'),
                input_content = '',
                output_content = '',
                root_spans = [
                    _generated.models.nested_span_with_metrics_response.NestedSpanWithMetricsResponse(
                        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'), )
                            ], 
                        children = [
                            _generated.models.nested_span_with_metrics_response.NestedSpanWithMetricsResponse(
                                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 = '', )
                            ], )
                    ],
                annotations = [
                    _generated.models.agentic_annotation_response.AgenticAnnotationResponse(
                        id = '', 
                        annotation_type = 'human', 
                        trace_id = '', 
                        continuous_eval_id = '', 
                        continuous_eval_name = '', 
                        eval_name = '', 
                        eval_version = 56, 
                        annotation_score = 56, 
                        annotation_description = '', 
                        input_variables = [
                            _generated.models.variable_template_value.VariableTemplateValue(
                                name = '', 
                                value = '', )
                            ], 
                        run_status = 'pending', 
                        cost = 1.337, 
                        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 TraceResponse(
                trace_id = '',
                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'),
        )
        """

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

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