add tests for expected_for method