def setup():
setup = Setup()
unb_segment = "UNB+UNOA:4+APIS*ABE+USADHS+070429:0900+000000001++USADHS'"
cc = Characters()
setup.cc = cc.with_control_character("decimal_point", ".")
setup.collection = RawSegmentCollection.from_str(unb_segment)
return setup
def test_split_by():
def _serialize(collections: Iterable[RawSegmentCollection]) -> List[List[str]]:
lst = []
global_lst = []
for collection in collections:
if lst:
global_lst.append(lst)
lst = []
for segment in collection.segments:
lst.append(segment.tag)
if lst:
global_lst.append(lst)
return global_lst
assert _serialize(RawSegmentCollection.from_segments([]).split_by("A")) == []
collection = RawSegmentCollection.from_segments(
Segment(i) for i in ["A", "B", "A", "A", "B", "D"]
)
assert _serialize(collection.split_by("Z")) == []
assert _serialize(collection.split_by("A")) == [["A", "B"], ["A"], ["A", "B", "D"]]
assert _serialize(collection.split_by("A")) == [["A", "B"], ["A"], ["A", "B", "D"]]
def test_get_segments_w_predicate():
collection = RawSegmentCollection.from_segments(
[
Segment("A", "1", "a"),
Segment("A", "2", "b"),
Segment("A", "1", "c"),
]
)
segments = collection.get_segments("A", lambda x: x[0] == "1")
assert [
Segment("A", "1", "a"),
Segment("A", "1", "c"),
] == list(segments)
def test_empty_segment_list():
m = RawSegmentCollection()
assert m.serialize() == ""
def test_get_segment_w_predicate():
collection = RawSegmentCollection.from_segments(
[Segment("36CF", "1"), Segment("36CF", "2")]
)
segment = collection.get_segment("36CF", lambda x: x[0] == "2")
assert segment == Segment("36CF", "2")
def test_get_segment():
collection = RawSegmentCollection.from_segments(
[Segment("36CF", 1), Segment("36CF", 2)]
)
segment = collection.get_segment("36CF")
assert Segment("36CF", 1) == segment
def test_get_segments():
collection = RawSegmentCollection.from_segments(
[Segment("36CF", 1), Segment("CPD"), Segment("36CF", 2)]
)
segments = list(collection.get_segments("36CF"))
assert [Segment("36CF", 1), Segment("36CF", 2)] == segments
def test_get_segments_doesnt_exist():
collection = RawSegmentCollection()
segments = list(collection.get_segments("36CF"))
assert [] == segments
def test_create_with_segments():
collection = RawSegmentCollection.from_segments([Segment("36CF")])
assert [Segment("36CF")] == collection.segments
def test_malformed_tag5():
with pytest.raises(EDISyntaxError):
RawSegmentCollection.from_str("IMD+F++:::This is '-:malformed string'")
def test_empty_segment():
m = RawSegmentCollection()
m.add_segment(Segment("", []))
assert m
def test_get_segment_doesnt_exist():
collection = RawSegmentCollection()
segment = collection.get_segment("36CF")
assert segment is None
def test_str_serialize():
collection = RawSegmentCollection.from_segments(
[Segment("36CF", "1"), Segment("36CF", "2")]
)
string = str(collection)
assert "36CF+1'36CF+2'" == string