Add more tests for stream