yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-11-07 05:53:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix imports

Browse Source
This commit is contained in:
Pablo Takara 2025-10-15 16:57:46 +02:00
parent dbae901260
commit bfbac4dca7
No known key found for this signature in database
GPG Key ID: 63381DDFBB2BF725
2 changed files with 4 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ingestion/src/metadata/data_quality/validations/base_test_handler.py
View File

@ -34,6 +34,7 @@ from pydantic import BaseModel
from metadata.data_quality.validations import utils from metadata.data_quality.validations import utils
from metadata.generated.schema.tests.basic import ( from metadata.generated.schema.tests.basic import (
DimensionValue,
TestCaseDimensionResult, TestCaseDimensionResult,
TestCaseResult, TestCaseResult,
TestCaseStatus, TestCaseStatus,
@ -462,8 +463,6 @@ class BaseTestValidator(ABC):
Returns: Returns:
DimensionResult: Dimension result object with calculated percentages DimensionResult: Dimension result object with calculated percentages
""" """
from metadata.generated.schema.tests.basic import DimensionValue
if failed_rows is None: if failed_rows is None:
failed_rows = total_rows - passed_rows failed_rows = total_rows - passed_rows

17
ingestion/tests/unit/test_suite/test_validations_databases.py
View File

@ -16,6 +16,7 @@ Each test should validate the Success, Failure and Aborted statuses
""" """
from datetime import date, datetime from datetime import date, datetime
from unittest.mock import patch from unittest.mock import patch
from uuid import uuid4
import pytest import pytest
@ -25,6 +26,8 @@ from metadata.data_quality.validations.models import (
from metadata.generated.schema.entity.services.databaseService import DatabaseConnection from metadata.generated.schema.entity.services.databaseService import DatabaseConnection
from metadata.generated.schema.tests.basic import TestCaseResult, TestCaseStatus from metadata.generated.schema.tests.basic import TestCaseResult, TestCaseStatus
from metadata.generated.schema.tests.testCase import TestCase, TestCaseParameterValue from metadata.generated.schema.tests.testCase import TestCase, TestCaseParameterValue
from metadata.generated.schema.type.entityReference import EntityReference
from metadata.profiler.metrics.registry import Metrics
from metadata.utils.importer import import_test_case_class from metadata.utils.importer import import_test_case_class
EXECUTION_DATE = datetime.strptime("2021-07-03", "%Y-%m-%d") EXECUTION_DATE = datetime.strptime("2021-07-03", "%Y-%m-%d")
@ -576,10 +579,6 @@ def test_suite_validation_database(
# Test cases for dimensional validation # Test cases for dimensional validation
def test_column_values_to_be_in_set_backward_compatibility(create_sqlite_table): def test_column_values_to_be_in_set_backward_compatibility(create_sqlite_table):
"""Test backward compatibility: non-dimensional ColumnValuesToBeInSet still works""" """Test backward compatibility: non-dimensional ColumnValuesToBeInSet still works"""
from uuid import uuid4
from metadata.generated.schema.type.entityReference import EntityReference
test_case = TestCase( test_case = TestCase(
name="test_dimensional_backward_compatibility", name="test_dimensional_backward_compatibility",
entityLink="<#E::table::service.db.users::columns::nickname>", entityLink="<#E::table::service.db.users::columns::nickname>",
@ -623,10 +622,6 @@ def test_column_values_to_be_in_set_backward_compatibility(create_sqlite_table):
def test_column_values_to_be_in_set_dimensional_validation(create_sqlite_table): def test_column_values_to_be_in_set_dimensional_validation(create_sqlite_table):
"""Test dimensional validation functionality""" """Test dimensional validation functionality"""
from uuid import uuid4
from metadata.generated.schema.type.entityReference import EntityReference
test_case = TestCase( test_case = TestCase(
id=uuid4(), id=uuid4(),
name="my_test_case", name="my_test_case",
@ -662,8 +657,6 @@ def test_column_values_to_be_in_set_dimensional_validation(create_sqlite_table):
# Mock the main query result and dimensional query results # Mock the main query result and dimensional query results
# Simulate realistic data: John and Jane are in the allowed set, Alice is not # Simulate realistic data: John and Jane are in the allowed set, Alice is not
# Note: Using Metrics enum names as keys after refactoring # Note: Using Metrics enum names as keys after refactoring
from metadata.profiler.metrics.registry import Metrics
mock_dimensional_data = [ mock_dimensional_data = [
{ {
"dimension_value": "John", "dimension_value": "John",
@ -763,10 +756,6 @@ def test_column_values_to_be_in_set_dimensional_validation(create_sqlite_table):
def test_column_values_to_be_in_set_invalid_dimension_column(create_sqlite_table): def test_column_values_to_be_in_set_invalid_dimension_column(create_sqlite_table):
"""Test error handling for invalid dimension columns""" """Test error handling for invalid dimension columns"""
from uuid import uuid4
from metadata.generated.schema.type.entityReference import EntityReference
test_case = TestCase( test_case = TestCase(
name="test_invalid_dimension_column", name="test_invalid_dimension_column",
entityLink="<#E::table::service.db.users::columns::nickname>", entityLink="<#E::table::service.db.users::columns::nickname>",