Correct yield statements in test_cyclus.py