def test_render_checkpoint_new_notebook_with_available_data_asset(
deterministic_asset_dataconnector_context,
titanic_expectation_suite,
checkpoint_new_notebook_assets,
):
"""
What does this test and why?
The CheckpointNewNotebookRenderer should generate a notebook with an example SimpleCheckpoint yaml config based on the first available data asset.
"""
context: DataContext = deterministic_asset_dataconnector_context
assert context.list_checkpoints() == []
context.save_expectation_suite(titanic_expectation_suite)
assert context.list_expectation_suite_names() == ["Titanic.warning"]
checkpoint_new_notebook_renderer = CheckpointNewNotebookRenderer(
context=context, checkpoint_name="my_checkpoint_name"
)
obs: nbformat.NotebookNode = checkpoint_new_notebook_renderer.render()
assert isinstance(obs, dict)
expected_cells = (
checkpoint_new_notebook_assets["header"]
+ checkpoint_new_notebook_assets["imports"]
+ checkpoint_new_notebook_assets[
"sample_checkpoint_config_markdown_description"
]
# Testing to make sure everything in the notebook but especially this checkpoint config code is correct.
+ checkpoint_new_notebook_assets["sample_checkpoint_config_code_correct"]
+ checkpoint_new_notebook_assets["optional_customize_your_config"]
+ checkpoint_new_notebook_assets["test_and_save_your_checkpoint_configuration"]
+ checkpoint_new_notebook_assets["review_checkpoint"]
+ checkpoint_new_notebook_assets["add_checkpoint"]
+ checkpoint_new_notebook_assets["optional_run_checkpoint"]
)
expected = {
"nbformat": 4,
"nbformat_minor": 4,
"metadata": {},
"cells": expected_cells,
}
del expected["nbformat_minor"]
del obs["nbformat_minor"]
for obs_cell, expected_cell in zip(obs["cells"], expected["cells"]):
obs_cell.pop("id", None)
assert obs_cell == expected_cell
assert obs == expected
def test_render_checkpoint_new_notebook_with_unavailable_data_asset(
assetless_dataconnector_context,
checkpoint_new_notebook_assets,
):
context: DataContext = assetless_dataconnector_context
assert context.list_checkpoints() == []
# This config is bad because of a missing expectation suite
checkpoint_new_notebook_renderer = CheckpointNewNotebookRenderer(
context=context, checkpoint_name="my_checkpoint_name"
)
obs: nbformat.NotebookNode = checkpoint_new_notebook_renderer.render()
assert isinstance(obs, dict)
expected_cells = (
checkpoint_new_notebook_assets["header"]
+ checkpoint_new_notebook_assets["imports"]
+ checkpoint_new_notebook_assets[
"sample_checkpoint_config_markdown_description"
]
# Testing to make sure the error message here is displayed appropriately
+ checkpoint_new_notebook_assets[
"sample_checkpoint_config_markdown_error_message"
]
+ checkpoint_new_notebook_assets["optional_customize_your_config"]
+ checkpoint_new_notebook_assets["test_and_save_your_checkpoint_configuration"]
+ checkpoint_new_notebook_assets["review_checkpoint"]
+ checkpoint_new_notebook_assets["add_checkpoint"]
+ checkpoint_new_notebook_assets["optional_run_checkpoint"]
)
expected = {
"nbformat": 4,
"nbformat_minor": 4,
"metadata": {},
"cells": expected_cells,
}
del expected["nbformat_minor"]
del obs["nbformat_minor"]
for obs_cell, expected_cell in zip(obs["cells"], expected["cells"]):
obs_cell.pop("id", None)
assert obs_cell == expected_cell
assert obs == expected