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_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_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_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]