Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: great expectations unit tests referring to deprecated method #1558

Merged
merged 8 commits into from
Mar 13, 2024
8 changes: 4 additions & 4 deletions tests/unit/test_ge_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def test_to_expectations_suite_context_save_and_build_data_docs(mod, context, df
mod.data_context.DataContext.assert_not_called()
mod.dataset.PandasDataset.assert_called_once()

context.create_expectation_suite.assert_called_once()
context.add_expectation_suite.assert_called_once()
context.save_expectation_suite.assert_called_once()
context.build_data_docs.assert_called_once()
context.open_data_docs.assert_called_once()
Expand All @@ -60,7 +60,7 @@ def test_to_expectations_suite_context_no_save_and_build_data_docs(mod, context,
mod.data_context.DataContext.assert_not_called()
mod.dataset.PandasDataset.assert_called_once()

context.create_expectation_suite.assert_called_once()
context.add_expectation_suite.assert_called_once()
context.save_expectation_suite.assert_called_once()
context.build_data_docs.assert_called_once()
context.open_data_docs.assert_called_once()
Expand All @@ -78,7 +78,7 @@ def test_to_expectations_suite_context_no_save_and_no_build_data_docs(mod, conte
mod.data_context.DataContext.assert_not_called()
mod.dataset.PandasDataset.assert_called_once()

context.create_expectation_suite.assert_called_once()
context.add_expectation_suite.assert_called_once()
context.save_expectation_suite.assert_not_called()
context.build_data_docs.assert_not_called()
context.open_data_docs.assert_not_called()
Expand All @@ -92,7 +92,7 @@ def test_to_expectations_suite_title(context, df):
run_validation=False,
)

context.create_expectation_suite.assert_called_once_with(
context.add_expectation_suite.assert_called_once_with(
"expectations-dataset", overwrite_existing=True
)

Expand Down
Loading