def test_expected(string: str, lookup: Dict = None, error: Exception = None) -> List[Recorded]:
messages = parse(
string,
timespan=timespan,
time_shift=subscribed,
lookup=lookup,
error=error,
)
return messages_to_records(messages)
def test_parse_marble_multiple_digits(self):
string = "-ab-cde--"
" 012345678901234567890"
results = parse(string)
expected = [
mess_on_next(1.0, 'ab'),
mess_on_next(4.0, 'cde'),
]
assert results == expected
def test_parse_marble_multiple_digits_int(self):
string = "-1-22-333-"
" 012345678901234567890"
results = parse(string)
expected = [
mess_on_next(1.0, 1),
mess_on_next(3.0, 22),
mess_on_next(6.0, 333),
]
assert results == expected
def test_parse_marble_completed(self):
string = "-ab-c--|"
" 012345678901234567890"
results = parse(string)
expected = [
mess_on_next(1.0, 'ab'),
mess_on_next(4.0, 'c'),
mess_on_completed(7.0),
]
assert results == expected
def test_parse_marble_multiple_digits_float(self):
string = "-1.0--2.345--6.7e8-"
" 012345678901234567890"
results = parse(string)
expected = [
mess_on_next(1.0, float('1.0')),
mess_on_next(6.0, float('2.345')),
mess_on_next(13.0, float('6.7e8')),
]
assert results == expected
def test_parse_marble_timedelta(self):
string = "a--b---c"
" 012345678901234567890"
ts = 0.1
results = parse(string, timespan=datetime.timedelta(seconds=ts))
expected = [
mess_on_next(0 * ts, 'a'),
mess_on_next(3 * ts, 'b'),
mess_on_next(7 * ts, 'c'),
]
assert results == expected
def test_parse_marble_with_space(self):
string = " -a b- c- de |"
" 01 23 45 67 8901234567890"
results = parse(string)
expected = [
mess_on_next(1.0, 'ab'),
mess_on_next(4.0, 'c'),
mess_on_next(6.0, 'de'),
mess_on_completed(8.0),
]
assert results == expected
def test_parse_marble_time_shift(self):
string = "-ab----c-d-|"
" 012345678901234567890"
offset = 10.0
results = parse(string, time_shift=offset)
expected = [
mess_on_next(1.0 + offset, 'ab'),
mess_on_next(7.0 + offset, 'c'),
mess_on_next(9.0 + offset, 'd'),
mess_on_completed(11.0 + offset),
]
assert results == expected
def test_parse_marble_with_error(self):
string = "-a-b-c--#--"
" 012345678901234567890"
ex = Exception('ex')
results = parse(string, error=ex)
expected = [
mess_on_next(1.0, 'a'),
mess_on_next(3.0, 'b'),
mess_on_next(5.0, 'c'),
mess_on_error(8.0, ex),
]
assert results == expected
def test_parse_marble_with_group(self):
string = "-x(ab,12,1.5)-c--(de)-|"
" 012345678901234567890123"
" 0 1 2 "
results = parse(string)
expected = [
mess_on_next(1.0, 'x'),
mess_on_next(2.0, 'ab'),
mess_on_next(2.0, 12),
mess_on_next(2.0, float('1.5')),
mess_on_next(14.0, 'c'),
mess_on_next(17.0, 'de'),
mess_on_completed(22.0),
]
assert results == expected
def test_parse_marble_with_group(self):
string = "-x(ab,12,1.5)-c--(de)-|"
" 012345678901234567890123"
" 0 1 2 "
results = parse(string)
expected = [
mess_on_next(1.0, 'x'),
mess_on_next(2.0, 'ab'),
mess_on_next(2.0, 12),
mess_on_next(2.0, float('1.5')),
mess_on_next(14.0, 'c'),
mess_on_next(17.0, 'de'),
mess_on_completed(22.0),
]
assert results == expected
def test_parse_marble_lookup(self):
string = "-ab-c-12-3-|"
" 012345678901234567890"
lookup = {
'ab': 'aabb',
'c': 'cc',
12: '1122',
3: 33,
}
results = parse(string, lookup=lookup)
expected = [
mess_on_next(1.0, 'aabb'),
mess_on_next(4.0, 'cc'),
mess_on_next(6.0, '1122'),
mess_on_next(9.0, 33),
mess_on_completed(11.0),
]
assert results == expected
def test_parse_marble_raise_with_elements_after_error_group(self):
string = "-a-b-c--(#,1)-"
" 012345678901234567890"
with self.assertRaises(ValueError):
parse(string, raise_stopped=True)
def test_parse_marble_raise_with_elements_after_completed(self):
string = "-a-b-c--|-1-"
" 012345678901234567890"
with self.assertRaises(ValueError):
parse(string, raise_stopped=True)
def test_parse_just_on_next(self):
string = "a"
results = parse(string)
expected = [mess_on_next(0.0, 'a')]
assert results == expected
def test_parse_just_on_completed(self):
string = "|"
results = parse(string)
expected = [mess_on_completed(0.0)]
assert results == expected
def test_parse_just_on_error_specified(self):
string = "#"
ex = Exception('Foo')
results = parse(string, error=ex)
expected = [mess_on_error(0.0, ex)]
assert results == expected
def test_parse_just_on_error(self):
string = "#"
results = parse(string)
expected = [mess_on_error(0.0, Exception('error'))]
assert results == expected
