Add tests for Stream