# 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 arthur_observability_sdk._generated.models.trace_list_response import TraceListResponse

class TestTraceListResponse(unittest.TestCase):
    """TraceListResponse unit test stubs"""

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def make_instance(self, include_optional) -> TraceListResponse:
        """Test TraceListResponse
            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 `TraceListResponse`
        """
        model = TraceListResponse()
        if include_optional:
            return TraceListResponse(
                count = 56,
                traces = [
                    arthur_observability_sdk._generated.models.trace_metadata_response.TraceMetadataResponse(
                        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 = '', 
                        task_id = '', 
                        user_id = '', 
                        session_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'), 
                        span_count = 56, 
                        duration_ms = 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'), 
                        input_content = '', 
                        output_content = '', 
                        annotations = [
                            arthur_observability_sdk._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 = [
                                    arthur_observability_sdk._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'), )
                            ], 
                        spans = [
                            arthur_observability_sdk._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 = arthur_observability_sdk._generated.models.raw_data.Raw Data(), 
                                input_content = '', 
                                output_content = '', 
                                metric_results = [
                                    arthur_observability_sdk._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 TraceListResponse(
                count = 56,
                traces = [
                    arthur_observability_sdk._generated.models.trace_metadata_response.TraceMetadataResponse(
                        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 = '', 
                        task_id = '', 
                        user_id = '', 
                        session_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'), 
                        span_count = 56, 
                        duration_ms = 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'), 
                        input_content = '', 
                        output_content = '', 
                        annotations = [
                            arthur_observability_sdk._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 = [
                                    arthur_observability_sdk._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'), )
                            ], 
                        spans = [
                            arthur_observability_sdk._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 = arthur_observability_sdk._generated.models.raw_data.Raw Data(), 
                                input_content = '', 
                                output_content = '', 
                                metric_results = [
                                    arthur_observability_sdk._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 testTraceListResponse(self):
        """Test TraceListResponse"""
        # inst_req_only = self.make_instance(include_optional=False)
        # inst_req_and_optional = self.make_instance(include_optional=True)

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