def test_wrong_polarization_data(self):
with pytest.raises(ValueError): # Only one.
Power(self.fh, polarization=['L'])
with pytest.raises(ValueError): # Duplication (same error as above)
Power(self.fh, polarization=['L', 'L', 'R', 'R'])
with pytest.raises(ValueError): # Wrong axis.
Power(self.fh, polarization=[['LL'], ['RR'], ['LR'], ['RL']])
def test_wrong_polarization_vdif(self):
with pytest.raises(AttributeError):
Power(self.fh)
fh = SetAttribute(self.fh,
polarization=np.array(['L', 'R'] * 4))
with pytest.raises(ValueError): # Too many.
Power(fh)
def test_frequency_sideband_mismatch(self):
frequency = np.array([[320.25], [320.25], [336.25], [336.25]]) * u.MHz
sideband = np.array([[-1], [1], [-1], [1]])
polarization = ['RR', 'LL', 'RL', 'LR']
# Create a fake stream a bit like the VDIF one, but with complex data.
bad_freq = np.array([[320, 320], [320, 320],
[336, 336], [336, 337]]) * u.MHz
eh = EmptyStreamGenerator((10000, 4, 2), sample_rate=1.*u.Hz,
start_time=Time('2018-01-01'),
frequency=bad_freq, sideband=sideband)
with pytest.raises(ValueError):
Power(eh, polarization=polarization)
bad_side = np.array([[-1, -1], [1, -1], [-1, -1], [1, 1]])
eh = EmptyStreamGenerator((10000, 4, 2), sample_rate=1.*u.Hz,
start_time=Time('2018-01-01'),
frequency=frequency, sideband=bad_side)
with pytest.raises(ValueError):
Power(eh, polarization=polarization)
def test_frequency_sideband_propagation(self):
# Regression test for gh-60
frequency = np.array([[320.25], [320.25], [336.25], [336.25]]) * u.MHz
sideband = np.array([[-1], [1], [-1], [1]])
polarization = ['R', 'L']
# Create a fake stream a bit like the VDIF one, but with complex data.
eh = EmptyStreamGenerator((10000, 4, 2), sample_rate=1.*u.Hz,
start_time=Time('2018-01-01'),
frequency=frequency, sideband=sideband,
polarization=polarization)
pt = Power(eh)
assert_array_equal(pt.polarization, np.array(['RR', 'LL', 'RL', 'LR']))
assert_array_equal(pt.frequency, eh.frequency)
assert_array_equal(pt.sideband, eh.sideband)
pt = Power(eh, polarization=pt.polarization)
assert_array_equal(pt.polarization, np.array(['RR', 'LL', 'RL', 'LR']))
assert_array_equal(pt.frequency, eh.frequency)
assert_array_equal(pt.sideband, eh.sideband)
def test_power(self):
# Load baseband file and get reference intensities.
fh = self.fh
ref_data = fh.read()
r0, i0, r1, i1 = ref_data.view('f4').T
ref_data = np.stack((r0 * r0 + i0 * i0,
r1 * r1 + i1 * i1,
r0 * r1 + i0 * i1,
i0 * r1 - r0 * i1), axis=1)
pt = Power(fh, polarization=['LL', 'RR', 'LR', 'RL'])
assert np.all(pt.polarization == np.array(['LL', 'RR', 'LR', 'RL']))
# Square everything.
data1 = pt.read()
assert abs(pt.time
- fh.start_time - fh.shape[0] / fh.sample_rate) < 1*u.ns
assert pt.dtype is ref_data.dtype is data1.dtype
assert np.allclose(ref_data, data1)
r = repr(pt)
assert r.startswith('Power(ih, polarization=')
pt.close()
def test_polarization_propagation(self):
# Add polarization information by hand.
fh = SetAttribute(self.fh,
polarization=np.array(['L', 'R']))
pt = Power(fh)
assert np.all(pt.polarization == np.array(['LL', 'RR', 'LR', 'RL']))
assert repr(pt).startswith('Power(ih)\n')
# Swap order.
fh2 = SetAttribute(self.fh,
polarization=np.array(['R', 'L']))
pt = Power(fh2)
assert np.all(pt.polarization == np.array(['RR', 'LL', 'RL', 'LR']))
pt.close()
def test_polarization_propagation2(self):
# Check it also works in other axes, or with an overly detailed array.
# Use a fake stream a bit like the VDIF one, but with complex data.
eh = EmptyStreamGenerator((10000, 2, 4), sample_rate=1.*u.Hz,
start_time=Time('2018-01-01'),
polarization=[['L'], ['R']])
pt = Power(eh)
expected = np.array([['LL'], ['RR'], ['LR'], ['RL']])
assert np.all(pt.polarization == expected)
pt = Power(eh, polarization=np.array([['LL'] * 4,
['RR'] * 4,
['LR'] * 4,
['RL'] * 4]))
assert np.all(pt.polarization == expected)
pt.close()
def test_power_needs_complex(self):
eh = EmptyStreamGenerator((10000, 2, 4), sample_rate=1.*u.Hz,
start_time=Time('2018-01-01'), dtype='f4',
polarization=[['L'], ['R']])
with pytest.raises(ValueError): # Real timestream
Power(eh)
def test_missing_polarization(self):
with pytest.raises(AttributeError):
Power(self.fh)