diff --git a/atrcopy/segments.py b/atrcopy/segments.py index 4f89723..403348b 100644 --- a/atrcopy/segments.py +++ b/atrcopy/segments.py @@ -553,7 +553,7 @@ class DefaultSegment: r = r.get_indexed[other.order] return r - def save_session(self, mdict): + def serialize_session(self, mdict): """Save extra metadata to a dict so that it can be serialized This is not saved by __getstate__ because child segments will point to diff --git a/test/test_serialize.py b/test/test_serialize.py index fea334e..dc145ee 100644 --- a/test/test_serialize.py +++ b/test/test_serialize.py @@ -47,15 +47,15 @@ class TestSegment: s.set_user_data([r], 4, 99) out = dict() - s.serialize_extra_to_dict(out) + s.serialize_session(out) print("saved", out) data = np.ones([4000], dtype=np.uint8) r = SegmentData(data) s2 = DefaultSegment(r, 0) - s2.restore_extra_from_dict(out) + s2.restore_session(out) out2 = dict() - s2.serialize_extra_to_dict(out2) + s2.serialize_session(out2) print("loaded", out2) assert out == out2