test_solver.py 文件源码

def runTest(self):
        """
        The first party of this tests to see if the solver routine 
        spits out any errors that it can raise internally. 
        We sequencially check the ucounts, bcounts,
        r, w, and cpars values.

        The second part checks some results. It compares the lam, p_mem, and wt
        values given by the solver to the values contained in the data file,
        which are precomputed solver outputs (from IDL).
        """
        file_name = 'test_solver_data.fit'
        file_path = 'data_for_tests'

        data=fitsio.read('%s/%s' % (file_path,file_name),ext=1)

        #need to transpose cpars
        data[0]['CPARS'] = data[0]['CPARS'][::-1]

        # check some common errors...
        testing.assert_raises(ValueError,redmapper.solver_nfw.Solver,data[0]['R0'],data[0]['BETA'],data[0]['UCOUNTS'][0:10],data[0]['BCOUNTS'],data[0]['R'],data[0]['W'],cpars=data[0]['CPARS'],rsig=data[0]['RSIG'])
        testing.assert_raises(ValueError,redmapper.solver_nfw.Solver,data[0]['R0'],data[0]['BETA'],data[0]['UCOUNTS'],data[0]['BCOUNTS'][0:10],data[0]['R'],data[0]['W'],cpars=data[0]['CPARS'],rsig=data[0]['RSIG'])
        testing.assert_raises(ValueError,redmapper.solver_nfw.Solver,data[0]['R0'],data[0]['BETA'],data[0]['UCOUNTS'],data[0]['BCOUNTS'],data[0]['R'][0:10],data[0]['W'],cpars=data[0]['CPARS'],rsig=data[0]['RSIG'])
        testing.assert_raises(ValueError,redmapper.solver_nfw.Solver,data[0]['R0'],data[0]['BETA'],data[0]['UCOUNTS'],data[0]['BCOUNTS'],data[0]['R'],data[0]['W'][0:10],cpars=data[0]['CPARS'],rsig=data[0]['RSIG'])
        testing.assert_raises(ValueError,redmapper.solver_nfw.Solver,data[0]['R0'],data[0]['BETA'],data[0]['UCOUNTS'],data[0]['BCOUNTS'],data[0]['R'],data[0]['W'],cpars=data[0]['CPARS'][0:1],rsig=data[0]['RSIG'])

        # and test the results
        solver=redmapper.solver_nfw.Solver(data[0]['R0'],data[0]['BETA'],data[0]['UCOUNTS'],data[0]['BCOUNTS'],data[0]['R'],data[0]['W'],cpars=data[0]['CPARS'],rsig=data[0]['RSIG'])

        """
        solve_nfw() spits out:
        lambda,
        p_mem,
        wt = p_mem*theta^L*theta^R
        """
        lam,p,wt,rlambda,theta_r=solver.solve_nfw()

        testing.assert_almost_equal(lam,data[0]['LAMBDA'])
        testing.assert_array_almost_equal(p,data[0]['PVALS'])
        testing.assert_array_almost_equal(wt,data[0]['WTVALS'])
        testing.assert_almost_equal(rlambda,data[0]['R0']*(data[0]['LAMBDA']/100.)**data[0]['BETA'],4)
        # need new test data with theta_r