Skip to content

Commit e3b2007

Browse files
committed
Add test for null DataStream edge case
1 parent b828f2a commit e3b2007

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

tests/test_contract/test_core.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from conftest import SimpleDataStream, SimpleParams
33

44
from contraqctor import _typing
5-
from contraqctor.contract.base import DataStreamCollection
5+
from contraqctor.contract.base import DataStream, DataStreamCollection
66

77

88
class 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

92104
class TestDataStreamCollection:
93105
"""Tests for the DataStreamCollection anonymous class."""

0 commit comments

Comments
 (0)