_TEST_FILTER_SPLIT_TRAINING = "train" _TEST_FILTER_SPLIT_VALIDATION = "validate" _TEST_FILTER_SPLIT_TEST = "test" _TEST_MODEL_NAME = ( f"projects/{_TEST_PROJECT}/locations/{_TEST_LOCATION}/models/{_TEST_MODEL_ID}" ) _TEST_PIPELINE_RESOURCE_NAME = ( f"projects/{_TEST_PROJECT}/locations/{_TEST_LOCATION}/trainingPipelines/12345" ) # CMEK encryption _TEST_DEFAULT_ENCRYPTION_KEY_NAME = "key_default" _TEST_DEFAULT_ENCRYPTION_SPEC = gca_encryption_spec.EncryptionSpec( kms_key_name=_TEST_DEFAULT_ENCRYPTION_KEY_NAME ) _TEST_PIPELINE_ENCRYPTION_KEY_NAME = "key_pipeline" _TEST_PIPELINE_ENCRYPTION_SPEC = gca_encryption_spec.EncryptionSpec( kms_key_name=_TEST_PIPELINE_ENCRYPTION_KEY_NAME ) _TEST_MODEL_ENCRYPTION_KEY_NAME = "key_model" _TEST_MODEL_ENCRYPTION_SPEC = gca_encryption_spec.EncryptionSpec( kms_key_name=_TEST_MODEL_ENCRYPTION_KEY_NAME ) @pytest.fixture def mock_pipeline_service_create():
# request_metadata _TEST_REQUEST_METADATA = () # dataset_metadata _TEST_NONTABULAR_DATASET_METADATA = None _TEST_METADATA_TABULAR_GCS = { "inputConfig": {"gcsSource": {"uri": [_TEST_SOURCE_URI_GCS]}} } _TEST_METADATA_TABULAR_BQ = { "inputConfig": {"bigquerySource": {"uri": _TEST_SOURCE_URI_BQ}} } # CMEK encryption _TEST_ENCRYPTION_KEY_NAME = "key_1234" _TEST_ENCRYPTION_SPEC = gca_encryption_spec.EncryptionSpec( kms_key_name=_TEST_ENCRYPTION_KEY_NAME ) # misc _TEST_OUTPUT_DIR = "gs://my-output-bucket" _TEST_DATASET_LIST = [ gca_dataset.Dataset( display_name="a", metadata_schema_uri=_TEST_METADATA_SCHEMA_URI_TABULAR ), gca_dataset.Dataset( display_name="d", metadata_schema_uri=_TEST_METADATA_SCHEMA_URI_NONTABULAR ), gca_dataset.Dataset( display_name="b", metadata_schema_uri=_TEST_METADATA_SCHEMA_URI_TABULAR ),