def test_get_polarization_tensor(self): ra = 1 dec = 2.0 time = 10 psi = 0.1 for mode in ["plus", "cross", "breathing", "longitudinal", "x", "y"]: p = gwutils.get_polarization_tensor(ra, dec, time, psi, mode) self.assertEqual(p.shape, (3, 3)) with self.assertRaises(ValueError): gwutils.get_polarization_tensor(ra, dec, time, psi, "not-a-mode")
def test_get_polarization_tensor(self): ra = 1 dec = 2.0 time = 10 psi = 0.1 for mode in ['plus', 'cross', 'breathing', 'longitudinal', 'x', 'y']: p = gwutils.get_polarization_tensor(ra, dec, time, psi, mode) self.assertEqual(p.shape, (3, 3)) with self.assertRaises(ValueError): gwutils.get_polarization_tensor(ra, dec, time, psi, 'not-a-mode')