from deploy.src.sagemaker_deploy_spec import SageMakerDeploySpec
import unittest


class DeploySpecTestCase(unittest.TestCase):

    REQUIRED_ARGS = ["--region", "us-west-2", "--model_name_1", "model-test"]

    def test_minimum_required_args(self):
        # Will raise if the inputs are incorrect
        spec = SageMakerDeploySpec(self.REQUIRED_ARGS)

from deploy.src.sagemaker_deploy_spec import SageMakerDeploySpec
import unittest


class DeploySpecTestCase(unittest.TestCase):

    REQUIRED_ARGS = ["--region", "us-west-2", "--model_name_1", "model-test"]

    def test_minimum_required_args(self):
        # Will raise if the inputs are incorrect
        spec = SageMakerDeploySpec(self.REQUIRED_ARGS)

# Will raise if the inputs are incorrect