Remove unused import
Add test for yield_transactions