def test_get_polarization_cutout_pixels_cgps():
"""
CTYPE4 = 'STOKES ' / 4TH COORDINATE TYPE
CRVAL4 = 1.000000000000E+00 / REF. COORD. 1-4=I,Q,U,V
CRPIX4 = 1.00 / REF. PIXEL
CDELT4 = 1.0000000E+00 / DELTA COORD.
CROTA4 = 0.00 / ROTATION ANGLE (DEG)
"""
header_filename = os.path.join(TESTDATA_DIR, CGPS_4D_CUBE_HEADER)
header = fits.Header.fromtextfile(header_filename)
polarization = Polarization([PolarizationState.I])
test_subject = Transform()
pixels = test_subject.get_polarization_cutout_pixels(
polarization, header, 4)
# SODA returns [*,*,*,1:1]
assert pixels is not None
assert len(pixels) == 2
assert pixels[0] == 1
assert pixels[1] == 1
def test_get_polarization_cutout_pixels_vlass():
"""
Polarization states for header are I, Q, U, V (1, 2, 3, 4)
"""
header_filename = os.path.join(TESTDATA_DIR, VLASS_4D_CUBE_HEADER)
header = fits.Header.fromtextfile(header_filename)
polarization = Polarization([PolarizationState.I])
test_subject = Transform()
pixels = test_subject.get_polarization_cutout_pixels(
polarization, header, 4)
# SODA returns [*,*,*,1:1]
assert pixels is not None
assert len(pixels) == 2
assert pixels[0] == 1
assert pixels[1] == 1
polarization = Polarization([PolarizationState.I, PolarizationState.Q])
pixels = test_subject.get_polarization_cutout_pixels(
polarization, header, 4)
# should return [1:2]
assert pixels is not None
assert len(pixels) == 2
assert pixels[0] == 1
assert pixels[1] == 2
polarization = Polarization(
[PolarizationState.I, PolarizationState.Q, PolarizationState.U])
pixels = test_subject.get_polarization_cutout_pixels(
polarization, header, 4)
# should return [1:3]
assert pixels is not None
assert len(pixels) == 2
assert pixels[0] == 1
assert pixels[1] == 3
polarization = Polarization([
PolarizationState.I, PolarizationState.Q, PolarizationState.U,
PolarizationState.V
])
pixels = test_subject.get_polarization_cutout_pixels(
polarization, header, 4)
# should return [1:4]
assert pixels is not None
assert len(pixels) == 2
assert pixels[0] == 1
assert pixels[1] == 4
polarization = Polarization([PolarizationState.I, PolarizationState.V])
pixels = test_subject.get_polarization_cutout_pixels(
polarization, header, 4)
# should return [1:4]
assert pixels is not None
assert len(pixels) == 2
assert pixels[0] == 1
assert pixels[1] == 4
