def test_save_load_permanent_marker_all_types(self, tmp_path, file):
filename = tmp_path / file
s = self.s
x1, y1, x2, y2 = 5, 2, 1, 8
m0_list = [
markers.point(x=x1, y=y1),
markers.horizontal_line(y=y1),
markers.horizontal_line_segment(x1=x1, x2=x2, y=y1),
markers.line_segment(x1=x1, x2=x2, y1=y1, y2=y2),
markers.rectangle(x1=x1, x2=x2, y1=y1, y2=y2),
markers.text(x=x1, y=y1, text="test"),
markers.vertical_line(x=x1),
markers.vertical_line_segment(x=x1, y1=y1, y2=y2),
]
for m in m0_list:
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
markers_dict = s1.metadata.Markers
m0_dict_list = []
m1_dict_list = []
for m in m0_list:
m0_dict_list.append(san_dict(m._to_dictionary()))
m1_dict_list.append(
san_dict(markers_dict.get_item(m.name)._to_dictionary()))
assert len(list(s1.metadata.Markers)) == 8
for m0_dict, m1_dict in zip(m0_dict_list, m1_dict_list):
assert m0_dict == m1_dict
def test_save_load_permanent_marker_all_types(self):
x1, y1, x2, y2 = 5, 2, 1, 8
s = Signal2D(np.arange(100).reshape(10, 10))
m0_list = [
markers.point(x=x1, y=y1),
markers.horizontal_line(y=y1),
markers.horizontal_line_segment(x1=x1, x2=x2, y=y1),
markers.line_segment(x1=x1, x2=x2, y1=y1, y2=y2),
markers.rectangle(x1=x1, x2=x2, y1=y1, y2=y2),
markers.text(x=x1, y=y1, text="test"),
markers.vertical_line(x=x1),
markers.vertical_line_segment(x=x1, y1=y1, y2=y2),
]
for m in m0_list:
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/testallmarkersfile.hdf5'
s.save(filename)
s1 = load(filename)
markers_dict = s1.metadata.Markers
m0_dict_list = []
m1_dict_list = []
for m in m0_list:
m0_dict_list.append(san_dict(m._to_dictionary()))
m1_dict_list.append(
san_dict(markers_dict.get_item(m.name)._to_dictionary()))
assert len(list(s1.metadata.Markers)) == 8
for m0_dict, m1_dict in zip(m0_dict_list, m1_dict_list):
assert m0_dict == m1_dict
def test_save_load_permanent_marker_all_types(self):
x1, y1, x2, y2 = 5, 2, 1, 8
s = Signal2D(np.arange(100).reshape(10, 10))
m0_list = [
markers.point(x=x1, y=y1),
markers.horizontal_line(y=y1),
markers.horizontal_line_segment(x1=x1, x2=x2, y=y1),
markers.line_segment(x1=x1, x2=x2, y1=y1, y2=y2),
markers.rectangle(x1=x1, x2=x2, y1=y1, y2=y2),
markers.text(x=x1, y=y1, text="test"),
markers.vertical_line(x=x1),
markers.vertical_line_segment(x=x1, y1=y1, y2=y2),
]
for m in m0_list:
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/testallmarkersfile.hdf5'
s.save(filename)
s1 = load(filename)
markers_dict = s1.metadata.Markers
m0_dict_list = []
m1_dict_list = []
for m in m0_list:
m0_dict_list.append(san_dict(m._to_dictionary()))
m1_dict_list.append(
san_dict(markers_dict.get_item(m.name)._to_dictionary()))
assert len(list(s1.metadata.Markers)) == 8
for m0_dict, m1_dict in zip(m0_dict_list, m1_dict_list):
assert m0_dict == m1_dict
def test_save_load_point_marker(self, tmp_path, file):
filename = tmp_path / file
s = self.s
x, y = 9, 8
color = 'purple'
name = "point test"
m = markers.point(x=x, y=y, color=color)
m.name = name
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_vertical_line_marker(self, tmp_path, file):
filename = tmp_path / file
s = self.s
x = 9
color = 'black'
linewidth = 3.5
name = "vertical_line_test"
m = markers.vertical_line(x=x, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_horizontal_line_marker(self, tmp_path, file):
filename = tmp_path / file
s = self.s
y = 8
color = 'blue'
linewidth = 2.5
name = "horizontal_line_test"
m = markers.horizontal_line(y=y, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_point_marker(self, mpl_cleanup):
x, y = 9, 8
color = 'purple'
name = "point test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.point(x=x, y=y, color=color)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_point_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_text_marker(self, tmp_path, file):
filename = tmp_path / file
s = self.s
x, y = 3, 9.5
color = 'brown'
name = "text_test"
text = "a text"
m = markers.text(x=x, y=y, text=text, color=color)
m.name = name
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_horizontal_line_marker(self, mpl_cleanup):
y = 8
color = 'blue'
linewidth = 2.5
name = "horizontal_line_test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.horizontal_line(y=y, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_horizontal_line_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_text_marker(self):
x, y = 3, 9.5
color = 'brown'
name = "text_test"
text = "a text"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.text(x=x, y=y, text=text, color=color)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_text_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_vertical_line_marker(self):
x = 9
color = 'black'
linewidth = 3.5
name = "vertical_line_test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.vertical_line(x=x, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_vertical_line_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_vertical_line_marker(self):
x = 9
color = 'black'
linewidth = 3.5
name = "vertical_line_test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.vertical_line(x=x, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_vertical_line_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_point_marker(self):
x, y = 9, 8
color = 'purple'
name = "point test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.point(
x=x, y=y, color=color)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_point_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_rectangle_marker(self, tmp_path, file):
filename = tmp_path / file
s = self.s
x1, x2, y1, y2 = 2, 4, 1, 3
color = 'yellow'
linewidth = 5
name = "rectangle_test"
m = markers.rectangle(
x1=x1, x2=x2, y1=y1, y2=y2, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_line_segment_marker(self, tmp_path, file):
filename = tmp_path / file
s = self.s
x1, x2, y1, y2 = 1, 9, 4, 7
color = 'cyan'
linewidth = 0.7
name = "line_segment_test"
m = markers.line_segment(
x1=x1, x2=x2, y1=y1, y2=y2, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_vertical_line_segment_marker(self, tmp_path, file):
filename = tmp_path / file
s = self.s
x, y1, y2 = 2, 1, 3
color = 'white'
linewidth = 4.2
name = "vertical_line_segment_test"
m = markers.vertical_line_segment(
x=x, y1=y1, y2=y2, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_horizontal_line_segment_marker(self, tmp_path, file):
filename = tmp_path / file
s = self.s
x1, x2, y = 1, 5, 8
color = 'red'
linewidth = 1.2
name = "horizontal_line_segment_test"
m = markers.horizontal_line_segment(
x1=x1, x2=x2, y=y, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_line_segment_marker(self, mpl_cleanup):
x1, x2, y1, y2 = 1, 9, 4, 7
color = 'cyan'
linewidth = 0.7
name = "line_segment_test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.line_segment(
x1=x1, x2=x2, y1=y1, y2=y2, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_line_segment_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_horizontal_line_segment_marker(self):
x1, x2, y = 1, 5, 8
color = 'red'
linewidth = 1.2
name = "horizontal_line_segment_test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.horizontal_line_segment(
x1=x1, x2=x2, y=y, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_horizontal_line_segment_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_rectangle_marker(self):
x1, x2, y1, y2 = 2, 4, 1, 3
color = 'yellow'
linewidth = 5
name = "rectangle_test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.rectangle(
x1=x1, x2=x2, y1=y1, y2=y2, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_rectangle_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_text_marker(self):
x, y = 3, 9.5
color = 'brown'
name = "text_test"
text = "a text"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.text(
x=x, y=y, text=text, color=color)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_text_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_rectangle_marker(self):
x1, x2, y1, y2 = 2, 4, 1, 3
color = 'yellow'
linewidth = 5
name = "rectangle_test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.rectangle(
x1=x1, x2=x2, y1=y1, y2=y2, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_rectangle_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_horizontal_line_segment_marker(self):
x1, x2, y = 1, 5, 8
color = 'red'
linewidth = 1.2
name = "horizontal_line_segment_test"
s = Signal2D(np.arange(100).reshape(10, 10))
m = markers.horizontal_line_segment(
x1=x1, x2=x2, y=y, color=color, linewidth=linewidth)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_horizontal_line_segment_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
def test_save_load_multidim_navigation_marker(self):
x, y = (1, 2, 3), (5, 6, 7)
name = 'test point'
s = Signal2D(np.arange(300).reshape(3, 10, 10))
m = markers.point(x=x, y=y)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_multidim_nav_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
assert m1.get_data_position('x1') == x[0]
assert m1.get_data_position('y1') == y[0]
s1.axes_manager.navigation_axes[0].index = 1
assert m1.get_data_position('x1') == x[1]
assert m1.get_data_position('y1') == y[1]
s1.axes_manager.navigation_axes[0].index = 2
assert m1.get_data_position('x1') == x[2]
assert m1.get_data_position('y1') == y[2]
def test_save_load_multidim_navigation_marker(self):
x, y = (1, 2, 3), (5, 6, 7)
name = 'test point'
s = Signal2D(np.arange(300).reshape(3, 10, 10))
m = markers.point(x=x, y=y)
m.name = name
s.add_marker(m, permanent=True)
with tempfile.TemporaryDirectory() as tmp:
filename = tmp + '/test_save_multidim_nav_marker.hdf5'
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
assert m1.get_data_position('x1') == x[0]
assert m1.get_data_position('y1') == y[0]
s1.axes_manager.navigation_axes[0].index = 1
assert m1.get_data_position('x1') == x[1]
assert m1.get_data_position('y1') == y[1]
s1.axes_manager.navigation_axes[0].index = 2
assert m1.get_data_position('x1') == x[2]
assert m1.get_data_position('y1') == y[2]
def test_save_load_multidim_navigation_marker(self, tmp_path, file, lazy):
filename = tmp_path / file
x, y = (1, 2, 3), (5, 6, 7)
name = 'test point'
s = Signal2D(np.arange(300).reshape(3, 10, 10))
if lazy:
s = s.as_lazy()
m = markers.point(x=x, y=y)
m.name = name
s.add_marker(m, permanent=True)
s.save(filename)
s1 = load(filename)
m1 = s1.metadata.Markers.get_item(name)
assert san_dict(m1._to_dictionary()) == san_dict(m._to_dictionary())
assert m1.get_data_position('x1') == x[0]
assert m1.get_data_position('y1') == y[0]
s1.axes_manager.navigation_axes[0].index = 1
assert m1.get_data_position('x1') == x[1]
assert m1.get_data_position('y1') == y[1]
s1.axes_manager.navigation_axes[0].index = 2
assert m1.get_data_position('x1') == x[2]
assert m1.get_data_position('y1') == y[2]
