def test_find_hot_dead_pixels():
"""Test hot and dead pixel searches"""
monitor = dark_monitor.Dark()
# Create "baseline" image
comparison_image = np.zeros((10, 10)) + 1.
# Create mean slope image to compare
mean_image = np.zeros((10, 10)) + 1.
mean_image[0, 0] = 1.7
mean_image[1, 1] = 2.2
mean_image[7, 7] = 4.5
mean_image[5, 5] = 0.12
mean_image[6, 6] = 0.06
mean_image[7, 3] = 0.09
hot, dead = monitor.find_hot_dead_pixels(mean_image,
comparison_image,
hot_threshold=2.,
dead_threshold=0.1)
assert len(hot) == 2
assert np.all(hot[0] == np.array([1, 7]))
assert np.all(hot[1] == np.array([1, 7]))
assert len(dead) == 2
assert np.all(dead[0] == np.array([6, 7]))
assert np.all(dead[1] == np.array([6, 3]))
def test_shift_to_full_frame():
"""Test pixel coordinate shifting to be in full frame coords"""
monitor = dark_monitor.Dark()
monitor.x0 = 512
monitor.y0 = 512
coordinates = (np.array([6, 7]), np.array([6, 3]))
new_coords = monitor.shift_to_full_frame(coordinates)
assert np.all(new_coords[0] == np.array([518, 519]))
assert np.all(new_coords[1] == np.array([518, 515]))
def test_get_metadata():
"""Test retrieval of metadata from input file"""
monitor = dark_monitor.Dark()
filename = os.path.join(get_config()['test_dir'], 'dark_monitor',
'test_image_1.fits')
monitor.get_metadata(filename)
assert monitor.detector == 'NRCA1'
assert monitor.x0 == 0
assert monitor.y0 == 0
assert monitor.xsize == 10
assert monitor.ysize == 10
assert monitor.sample_time == 10
assert monitor.frame_time == 10.5
def test_noise_check():
"""Test the search for noisier than average pixels"""
noise_image = np.zeros((10, 10)) + 0.5
baseline = np.zeros((10, 10)) + 0.5
noise_image[3, 3] = 0.8
noise_image[6, 6] = 0.6
noise_image[9, 9] = 1.0
baseline[5, 5] = 1.0
noise_image[5, 5] = 1.25
monitor = dark_monitor.Dark()
noisy = monitor.noise_check(noise_image, baseline, threshold=1.5)
assert len(noisy[0]) == 2
assert np.all(noisy[0] == np.array([3, 9]))
assert np.all(noisy[1] == np.array([3, 9]))