# 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.rag_test_case import RagTestCase

class TestRagTestCase(unittest.TestCase):
    """RagTestCase unit test stubs"""

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def make_instance(self, include_optional) -> RagTestCase:
        """Test RagTestCase
            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 `RagTestCase`
        """
        model = RagTestCase()
        if include_optional:
            return RagTestCase(
                status = 'queued',
                dataset_row_id = '',
                total_cost = '',
                rag_results = [
                    arthur_observability_sdk._generated.models.rag_result.RagResult(
                        evals = [
                            arthur_observability_sdk._generated.models.eval_execution.EvalExecution(
                                eval_name = '', 
                                eval_version = '', 
                                eval_input_variables = [
                                    arthur_observability_sdk._generated.models.input_variable.InputVariable(
                                        variable_name = '', 
                                        value = '', )
                                    ], 
                                eval_results = arthur_observability_sdk._generated.models.eval_execution_result.EvalExecutionResult(
                                    score = 1.337, 
                                    explanation = '', 
                                    cost = '', ), )
                            ], 
                        rag_config_key = '', 
                        rag_config_type = '', 
                        setting_configuration_id = '', 
                        setting_configuration_version = 56, 
                        query_text = '', 
                        output = arthur_observability_sdk._generated.models.rag_search_output.RagSearchOutput(
                            response = arthur_observability_sdk._generated.models.rag_provider_query_response.RagProviderQueryResponse(
                                response = arthur_observability_sdk._generated.models.weaviate_query_results.WeaviateQueryResults(
                                    rag_provider = 'weaviate', 
                                    objects = [
                                        arthur_observability_sdk._generated.models.weaviate_query_result.WeaviateQueryResult(
                                            uuid = '', 
                                            metadata = arthur_observability_sdk._generated.models.weaviate_query_result_metadata.WeaviateQueryResultMetadata(
                                                creation_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                                                last_update_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                                                distance = 1.337, 
                                                certainty = 1.337, 
                                                score = 1.337, 
                                                explain_score = '', 
                                                is_consistent = True, ), 
                                            properties = arthur_observability_sdk._generated.models.properties.Properties(), 
                                            vector = {
                                                'key' : null
                                                }, )
                                        ], ), ), ), )
                    ]
            )
        else:
            return RagTestCase(
                status = 'queued',
                dataset_row_id = '',
                rag_results = [
                    arthur_observability_sdk._generated.models.rag_result.RagResult(
                        evals = [
                            arthur_observability_sdk._generated.models.eval_execution.EvalExecution(
                                eval_name = '', 
                                eval_version = '', 
                                eval_input_variables = [
                                    arthur_observability_sdk._generated.models.input_variable.InputVariable(
                                        variable_name = '', 
                                        value = '', )
                                    ], 
                                eval_results = arthur_observability_sdk._generated.models.eval_execution_result.EvalExecutionResult(
                                    score = 1.337, 
                                    explanation = '', 
                                    cost = '', ), )
                            ], 
                        rag_config_key = '', 
                        rag_config_type = '', 
                        setting_configuration_id = '', 
                        setting_configuration_version = 56, 
                        query_text = '', 
                        output = arthur_observability_sdk._generated.models.rag_search_output.RagSearchOutput(
                            response = arthur_observability_sdk._generated.models.rag_provider_query_response.RagProviderQueryResponse(
                                response = arthur_observability_sdk._generated.models.weaviate_query_results.WeaviateQueryResults(
                                    rag_provider = 'weaviate', 
                                    objects = [
                                        arthur_observability_sdk._generated.models.weaviate_query_result.WeaviateQueryResult(
                                            uuid = '', 
                                            metadata = arthur_observability_sdk._generated.models.weaviate_query_result_metadata.WeaviateQueryResultMetadata(
                                                creation_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                                                last_update_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                                                distance = 1.337, 
                                                certainty = 1.337, 
                                                score = 1.337, 
                                                explain_score = '', 
                                                is_consistent = True, ), 
                                            properties = arthur_observability_sdk._generated.models.properties.Properties(), 
                                            vector = {
                                                'key' : null
                                                }, )
                                        ], ), ), ), )
                    ],
        )
        """

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

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