def test_known_values(self): ''' polarisation_tensor_basis should give known result ''' for xy, tensors in self.known_values: x, y = xy expected_eplus, expected_ecros = tensors eplus = ETwaves.polarisation_tensor_basis(x, y, pol = '+') ecros = ETwaves.polarisation_tensor_basis(x, y, pol = 'x') self.assertTrue(np.allclose(expected_eplus, eplus)) self.assertTrue(np.allclose(expected_ecros, ecros))