OpenMetadata/ingestion/tests/unit/profiler/test_profiler_models.py
2025-04-03 10:39:47 +05:30

26 lines
1003 B
Python

# Copyright 2025 Collate
# Licensed under the Collate Community License, Version 1.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# https://github.com/open-metadata/OpenMetadata/blob/main/ingestion/LICENSE
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Profiler models behave properly"""
import pytest
from metadata.profiler.processor.models import ProfilerDef
def test_valid_metrics():
"""
Test that the metrics are valid
"""
profiler_def = ProfilerDef(name="test", metrics=["count"])
assert profiler_def.metrics == ["COUNT"]
with pytest.raises(ValueError):
ProfilerDef(name="test", metrics=["potato"])