Add tests to streams_tests.py