kopia lustrzana https://github.com/bugout-dev/moonstream
57 wiersze
1.8 KiB
Python
57 wiersze
1.8 KiB
Python
"""
|
|
Tests for stream boundary utilities.
|
|
"""
|
|
import unittest
|
|
|
|
from . import stream_boundaries
|
|
from .data import StreamBoundary
|
|
|
|
|
|
class TestValidateStreamBoundary(unittest.TestCase):
|
|
def test_valid_stream_boundary(self):
|
|
stream_boundary = StreamBoundary(
|
|
start_time=1, end_time=5, include_start=True, include_end=True
|
|
)
|
|
valid, _ = stream_boundaries.validate_stream_boundary(
|
|
stream_boundary, 10, raise_when_invalid=False
|
|
)
|
|
self.assertTrue(valid)
|
|
|
|
def test_invalid_stream_boundary(self):
|
|
stream_boundary = StreamBoundary(
|
|
start_time=1, end_time=5, include_start=True, include_end=True
|
|
)
|
|
valid, _ = stream_boundaries.validate_stream_boundary(
|
|
stream_boundary, 1, raise_when_invalid=False
|
|
)
|
|
self.assertFalse(valid)
|
|
|
|
def test_invalid_stream_boundary_error(self):
|
|
stream_boundary = StreamBoundary(
|
|
start_time=1, end_time=5, include_start=True, include_end=True
|
|
)
|
|
valid, _ = stream_boundaries.validate_stream_boundary(
|
|
stream_boundary, 1, raise_when_invalid=True
|
|
)
|
|
with self.assertRaises(stream_boundaries.InvalidStreamBoundary):
|
|
valid
|
|
|
|
def test_unconstrainted_invalid_stream_boundary(self):
|
|
stream_boundary = StreamBoundary()
|
|
valid, _ = stream_boundaries.validate_stream_boundary(
|
|
stream_boundary, 1, raise_when_invalid=False
|
|
)
|
|
self.assertFalse(valid)
|
|
|
|
def test_unconstrained_invalid_stream_boundary_error(self):
|
|
stream_boundary = StreamBoundary()
|
|
valid, _ = stream_boundaries.validate_stream_boundary(
|
|
stream_boundary, 1, raise_when_invalid=True
|
|
)
|
|
with self.assertRaises(stream_boundaries.InvalidStreamBoundary):
|
|
valid
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|