diff --git a/test/test_spaces/test_power_space.py b/test/test_spaces/test_power_space.py index 4e8951ab9fb178010d0d834e7857ca42cd2f9087..a791b6388db84ba3eafe56da3f1dc418bf8506a5 100644 --- a/test/test_spaces/test_power_space.py +++ b/test/test_spaces/test_power_space.py @@ -114,11 +114,18 @@ class PowerSpaceConsistencyCheck(unittest.TestCase): def test_pipundexInversion(self, harmonic_partner, distribution_strategy, logarithmic, nbin, binbounds, expected): #expected will not be used TODO: write expandproduct to use for this - p = PowerSpace(harmonic_partner=harmonic_partner, + if 'error' in expected: + with assert_raises(expected['error']): + PowerSpace(harmonic_partner=harmonic_partner, + distribution_strategy=distribution_strategy, + logarithmic=logarithmic, nbin=nbin, + binbounds=binbounds) + else: + p = PowerSpace(harmonic_partner=harmonic_partner, distribution_strategy=distribution_strategy, logarithmic=logarithmic, nbin=nbin, binbounds=binbounds) - assert_equal(p.pindex[p.pundex],np.arange(p.dim),err_msg='pundex is not right-inverse of pindex!') + assert_equal(p.pindex[p.pundex],np.arange(p.dim),err_msg='pundex is not right-inverse of pindex!') class PowerSpaceFunctionalityTest(unittest.TestCase): @expand(CONSTRUCTOR_CONFIGS)