"""Unit tests for dataflow modules."""