def test_WavesDir(self):
# Test the error on wrong input type
with self.assertRaises(TypeError):
cw.GravitationalWavesDir(0)
with self.assertRaises(TypeError):
cw.ElectromagneticWavesDir(0)
emdir = cw.ElectromagneticWavesDir(self.sim)
# Check type
self.assertTrue(
isinstance(self.gwdir[110.69], cw.GravitationalWavesOneDet))
self.assertTrue(
isinstance(emdir[110.69], cw.ElectromagneticWavesOneDet))
def electromagneticwaves(self):
"""Return all the available ``Phi2`` data.
:returns: Interface to all the ``Phi2`` data in the directory.
:rtype: :py:class:`~.ElectromagneticWavesDir`
"""
return cactus_waves.ElectromagneticWavesDir(self)
def test_get_power_energy_em(self):
emdir = cw.ElectromagneticWavesDir(self.sim)
phi2 = emdir[110.69]
phi2lm = phi2[(2, 2)]
power_lm = phi2.dist**2 / (4 * np.pi) * np.abs(phi2lm)**2
self.assertEqual(power_lm, phi2.get_power_lm(2, 2))
self.assertEqual(power_lm.integrated(), phi2.get_energy_lm(2, 2))
# Total power
total_power = (phi2.get_power_lm(2, 2) + phi2.get_power_lm(2, 1) +
phi2.get_power_lm(2, 0) + phi2.get_power_lm(2, -1) +
phi2.get_power_lm(2, -2))
self.assertEqual(total_power, phi2.get_total_power())
self.assertEqual(total_power.integrated(), phi2.get_total_energy())