feat(SIMPL-24642): migrate tests from 3 source repos with updated imports
This commit is contained in:
53
tests/data_processing/conftest.py
Normal file
53
tests/data_processing/conftest.py
Normal file
@@ -0,0 +1,53 @@
|
||||
"""Pytest configuration and shared fixtures."""
|
||||
|
||||
import pytest
|
||||
import pandas as pd
|
||||
from unittest.mock import MagicMock, patch
|
||||
import sys
|
||||
from dagster import build_op_context
|
||||
|
||||
# Mock external dependencies that might not be available in test environment
|
||||
sys.modules['spellchecker'] = MagicMock()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_context():
|
||||
"""Create a mock Dagster context for testing operations."""
|
||||
context = build_op_context()
|
||||
return context
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def sample_dataframe():
|
||||
"""Create a sample DataFrame for testing."""
|
||||
return pd.DataFrame({
|
||||
'Name': ['John Doe', 'jane smith', 'John Doe', 'bob johnson', 'John Doe'],
|
||||
'Age': [25, 30, 25, None, 25],
|
||||
'City': ['New York', 'los angeles', 'New York', 'chicago', 'New York'],
|
||||
'Status': ['Active', 'INACTIVE', 'Active', 'penDing', 'Active']
|
||||
})
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def sample_dataframe_with_typos():
|
||||
"""Create a sample DataFrame with typos for spell checking."""
|
||||
return pd.DataFrame({
|
||||
'Name': ['jon doe', 'jane smith', 'bob jonson'],
|
||||
'Description': ['developer', 'analst', 'enginer']
|
||||
})
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def empty_dataframe():
|
||||
"""Create an empty DataFrame."""
|
||||
return pd.DataFrame()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def dataframe_with_missing_values():
|
||||
"""Create a DataFrame with various missing values."""
|
||||
return pd.DataFrame({
|
||||
'Column1': [1, None, 3, None, 5],
|
||||
'Column2': ['a', 'b', None, 'd', None],
|
||||
'Column3': [None, None, None, None, None]
|
||||
})
|
||||
Reference in New Issue
Block a user