def test_get_classification_schema_location_returns_correct_for_prod():
file_name = classify.get_classification_schema_location(
Environments.PROD, 'classif_lead')
assert 's3://jornaya-prod-us-east-1-prj/classification/classif_lead/' == file_name
def test_get_classification_schema_location_returns_correct_for_local():
file_name = classify.get_classification_schema_location(
Environments.LOCAL, 'classif_lead')
assert '../samples/classification/classif_lead/' == file_name
def test_get_classification_schema_location_returns_correct_for_staging():
file_name = classify.get_classification_schema_location(
Environments.STAGING, 'classif_lead')
assert 's3://jornaya-staging-us-east-1-prj/classification/classif_lead/' == file_name