1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
|
import unittest
import reactivex
from reactivex.testing import ReactiveTest, TestScheduler
on_next = ReactiveTest.on_next
on_completed = ReactiveTest.on_completed
on_error = ReactiveTest.on_error
subscribe = ReactiveTest.subscribe
subscribed = ReactiveTest.subscribed
disposed = ReactiveTest.disposed
created = ReactiveTest.created
class RxException(Exception):
pass
# Helper function for raising exceptions within lambdas
def _raise(ex):
raise RxException(ex)
class TestTimeInterval(unittest.TestCase):
def test_interval_timespan_basic(self):
scheduler = TestScheduler()
def create():
return reactivex.interval(100)
results = scheduler.start(create)
assert results.messages == [
on_next(300, 0),
on_next(400, 1),
on_next(500, 2),
on_next(600, 3),
on_next(700, 4),
on_next(800, 5),
on_next(900, 6),
]
# def test_interval_timespan_zero(self):
# scheduler = TestScheduler()
# def create():
# return reactivex.interval(0)
# results = scheduler.start(create, disposed=210)
# assert results.messages == [
# on_next(201, 0), on_next(202, 1), on_next(203, 2),
# on_next(204, 3), on_next(205, 4), on_next(206, 5),
# on_next(207, 6), on_next(208, 7), on_next(209, 8)]
# def test_interval_timespan_negative(self):
# scheduler = TestScheduler()
# def create():
# return reactivex.interval(-1)
# results = scheduler.start(create, disposed=210)
# assert results.messages == [
# on_next(201, 0), on_next(202, 1), on_next(203, 2),
# on_next(204, 3), on_next(205, 4), on_next(206, 5),
# on_next(207, 6), on_next(208, 7), on_next(209, 8)]
def test_interval_timespan_disposed(self):
scheduler = TestScheduler()
def create():
return reactivex.interval(1000)
results = scheduler.start(create)
assert results.messages == []
def test_interval_timespan_observer_throws(self):
scheduler = TestScheduler()
xs = reactivex.interval(1)
xs.subscribe(lambda x: _raise("ex"), scheduler=scheduler)
with self.assertRaises(RxException):
scheduler.start()
|