File tree Expand file tree Collapse file tree 1 file changed +13
-1
lines changed
Expand file tree Collapse file tree 1 file changed +13
-1
lines changed Original file line number Diff line number Diff line change 22from conftest import SimpleDataStream , SimpleParams
33
44from contraqctor import _typing
5- from contraqctor .contract .base import DataStreamCollection
5+ from contraqctor .contract .base import DataStream , DataStreamCollection
66
77
88class TestDataStream :
@@ -88,6 +88,18 @@ def test_clear_data(self, text_file):
8888 with pytest .raises (ValueError ):
8989 _ = stream .data # Accessing data after clearing should raise ValueError
9090
91+ def test_null_data_stream (self ):
92+ """Test DataStream with None data type."""
93+
94+ class _NullDataStream (DataStream [None , None ]):
95+ @staticmethod
96+ def _reader (params : None = None ) -> None :
97+ return None
98+
99+ null_stream = _NullDataStream (name = "null_stream" , description = "Null data stream" , reader_params = None )
100+ null_stream .load ()
101+ assert null_stream .data is None
102+
91103
92104class TestDataStreamCollection :
93105 """Tests for the DataStreamCollection anonymous class."""
You can’t perform that action at this time.
0 commit comments