# 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.set_notebook_state_request import SetNotebookStateRequest

class TestSetNotebookStateRequest(unittest.TestCase):
    """SetNotebookStateRequest unit test stubs"""

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def make_instance(self, include_optional) -> SetNotebookStateRequest:
        """Test SetNotebookStateRequest
            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 `SetNotebookStateRequest`
        """
        model = SetNotebookStateRequest()
        if include_optional:
            return SetNotebookStateRequest(
                state = _generated.models.notebook_state.NotebookState(
                    prompt_configs = [
                        null
                        ], 
                    prompt_variable_mapping = [
                        _generated.models.prompt_variable_mapping.PromptVariableMapping(
                            variable_name = '', 
                            source = _generated.models.dataset_column_variable_source.DatasetColumnVariableSource(
                                type = 'dataset_column', 
                                dataset_column = _generated.models.dataset_column_source.DatasetColumnSource(
                                    name = '', ), ), )
                        ], 
                    dataset_ref = _generated.models.dataset_ref.DatasetRef(
                        id = '', 
                        name = '', 
                        version = 56, ), 
                    dataset_row_filter = [
                        _generated.models.new_dataset_version_row_column_item_request.NewDatasetVersionRowColumnItemRequest(
                            column_name = '', 
                            column_value = '', )
                        ], 
                    eval_list = [
                        _generated.models.eval_ref.EvalRef(
                            name = '', 
                            version = 56, 
                            variable_mapping = [
                                _generated.models.eval_variable_mapping.EvalVariableMapping(
                                    variable_name = '', 
                                    source = null, )
                                ], )
                        ], )
            )
        else:
            return SetNotebookStateRequest(
                state = _generated.models.notebook_state.NotebookState(
                    prompt_configs = [
                        null
                        ], 
                    prompt_variable_mapping = [
                        _generated.models.prompt_variable_mapping.PromptVariableMapping(
                            variable_name = '', 
                            source = _generated.models.dataset_column_variable_source.DatasetColumnVariableSource(
                                type = 'dataset_column', 
                                dataset_column = _generated.models.dataset_column_source.DatasetColumnSource(
                                    name = '', ), ), )
                        ], 
                    dataset_ref = _generated.models.dataset_ref.DatasetRef(
                        id = '', 
                        name = '', 
                        version = 56, ), 
                    dataset_row_filter = [
                        _generated.models.new_dataset_version_row_column_item_request.NewDatasetVersionRowColumnItemRequest(
                            column_name = '', 
                            column_value = '', )
                        ], 
                    eval_list = [
                        _generated.models.eval_ref.EvalRef(
                            name = '', 
                            version = 56, 
                            variable_mapping = [
                                _generated.models.eval_variable_mapping.EvalVariableMapping(
                                    variable_name = '', 
                                    source = null, )
                                ], )
                        ], ),
        )
        """

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

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