# Testing Prefect flows and tasks

Looking for the latest Prefect 2 release? Prefect 2 and Prefect Cloud 2 have been released for General Availability. See https://docs.prefect.io/ for details.

Since Prefect flows are Python objects they can be tested in any way you would normally test your Python code! This means that it is common for users to test their flows in unit tests with libraries like pytest. Below runs through a few ways to test your flows and if you want further examples Prefect's tests directory has thousands of tests that could serve as inspiration for testing your flows and tasks.

Use the following flow as an example:

# Testing flow composition

from prefect.core.edge import Edge

assert e in flow.tasks
assert t in flow.tasks
assert l in flow.tasks
assert len(flow.tasks) == 3

assert flow.root_tasks() == set([e])
assert flow.terminal_tasks() == set([l])

assert len(flow.edges) == 2
assert Edge(upstream_task=e, downstream_task=t, key="x") in flow.edges
assert Edge(upstream_task=t, downstream_task=l, key="x") in flow.edges

# Testing state

state = flow.run()

assert state.is_successful()
assert state.result[e].is_successful()
assert state.result[t].is_successful()
assert state.result[l].is_successful()

# Testing results

state = flow.run()

assert state.result[e].result == 10
assert state.result[t].result == 20
assert state.result[l].result == None