Skip to content

Commit

Permalink
test: add unit tests for retry_policy in decorators
Browse files Browse the repository at this point in the history
  • Loading branch information
adam-bloom committed Oct 21, 2022
1 parent 2d44525 commit 27a44e8
Showing 1 changed file with 27 additions and 0 deletions.
Expand Up @@ -22,6 +22,7 @@
build_assets_job,
multi_asset,
)
from dagster._core.definitions.policy import RetryPolicy
from dagster._core.definitions.resource_requirement import ensure_requirements_satisfied
from dagster._core.errors import DagsterInvalidConfigError
from dagster._core.types.dagster_type import resolve_dagster_type
Expand Down Expand Up @@ -583,3 +584,29 @@ def other_asset():
# If IO manager def is provided as a resource def, it appears in required
# resource keys on the underlying op.
assert set(other_asset.node_def.required_resource_keys) == {"blah"}


def test_asset_retry_policy():
retry_policy = RetryPolicy()

@asset(retry_policy=retry_policy)
def my_asset():
...

assert my_asset.op.retry_policy == retry_policy


def test_multi_asset_retry_policy():
retry_policy = RetryPolicy()

@multi_asset(
outs={
"key1": Out(asset_key=AssetKey("key1")),
"key2": Out(asset_key=AssetKey("key2")),
},
retry_policy=retry_policy,
)
def my_asset():
...

assert my_asset.op.retry_policy == retry_policy

0 comments on commit 27a44e8

Please sign in to comment.