def test_get_circle_mask(self): mask = sr.get_circle_mask(map_alt=self.monitor.deg_coord_y, map_azi=self.monitor.deg_coord_x, center=(10., 60.), radius=20., is_smooth_edge=True, blur_ratio=0.5, blur_func=sr.blur_cos, is_plot=False) # print mask[39, 100] assert (mask[39, 100] - 0.404847 < 1E10)
def test_get_circle_mask(self): mask = sr.get_circle_mask(map_alt=self.monitor.deg_coord_y, map_azi=self.monitor.deg_coord_x, center=(10., 60.), radius=20., is_smooth_edge=True, blur_ratio=0.5, blur_func=sr.blur_cos, is_plot=False) # print mask[39, 100] assert (mask[39, 100] - 0.404847 < 1E10)
def test_get_circle_mask2(self): import numpy as np alt = np.arange(-30., 30., 1.)[::-1] azi = np.arange(-30., 30., 1.) azi_map, alt_map = np.meshgrid(azi, alt) cm = sr.get_circle_mask(map_alt=alt_map, map_azi=azi_map, center=(0., 10.), radius=10., is_smooth_edge=False) # import matplotlib.pyplot as plt # plt.imshow(cm) # plt.show() assert (cm[28, 49] == 1) cm = sr.get_circle_mask(map_alt=alt_map, map_azi=azi_map, center=(10., 0.), radius=10., is_smooth_edge=False) # import matplotlib.pyplot as plt # plt.imshow(cm) # plt.show() assert (cm[10, 30] == 1)
def test_get_circle_mask2(self): import numpy as np alt = np.arange(-30., 30., 1.)[::-1] azi = np.arange(-30., 30., 1.) azi_map, alt_map = np.meshgrid(azi, alt) cm = sr.get_circle_mask(map_alt=alt_map, map_azi=azi_map, center=(0., 10.), radius=10., is_smooth_edge=False) # import matplotlib.pyplot as plt # plt.imshow(cm) # plt.show() assert (cm[28, 49] == 1) cm = sr.get_circle_mask(map_alt=alt_map, map_azi=azi_map, center=(10., 0.), radius=10., is_smooth_edge=False) # import matplotlib.pyplot as plt # plt.imshow(cm) # plt.show() assert (cm[10, 30] == 1)