Exemplo n.º 1
0
def test_starfit():
    rootdir = os.path.abspath(os.path.join(os.path.dirname(__file__)))
    testdir = os.path.join(rootdir, "star1")
    if mnest:
        basename = "{}/{}-".format(chainsdir, np.random.randint(1000000))
        kwargs = dict(n_live_points=20,
                      max_iter=100,
                      basename=basename,
                      verbose=False)
        getLogger().info("Testing starfit function with multinest...")
    else:
        kwargs = dict(nburn=20, niter=20, ninitial=10)
        getLogger().info("Testing starfit function with emcee...")

    mod, _ = starfit(testdir,
                     overwrite=True,
                     use_emcee=not mnest,
                     no_plots=True,
                     **kwargs)

    mod.samples

    if mnest:
        files = glob.glob("{}*".format(basename))
        for f in files:
            os.remove(f)
Exemplo n.º 2
0
    def getAll(self):
        self._createIni()
        kwargs = self._createKwargs()
        self.starModel, _ = starfit('.', **kwargs)
        self._clean()

        self.samples = self.starModel.samples
        resume = self.samples.describe()

        self.mass = round(resume.at['mean', 'mass_0_0'], 2)
        self.masserr = round(resume.at['std', 'mass_0_0'], 2)
        self.radius = round(resume.at['mean', 'radius_0_0'], 2)
        self.radiuserr = round(resume.at['std', 'radius_0_0'], 2)
        self.age = round(resume.at['mean', 'age_0_0'], 2)
        self.ageerr = round(resume.at['std', 'age_0_0'], 2)
Exemplo n.º 3
0
def test_starfit():
    rootdir = os.path.abspath(os.path.join(os.path.dirname(__file__)))
    testdir = os.path.join(rootdir, 'star1')
    if mnest:
        basename = '{}/{}-'.format(chainsdir,np.random.randint(1000000))
        kwargs = dict(n_live_points=20, max_iter=100,basename=basename,
                      verbose=False)
    else:
        kwargs = dict(nburn=20, niter=20, ninitial=10)

    mod, _ = starfit(testdir, overwrite=True, use_emcee=not mnest,
                     no_plots=True, **kwargs)

    mod.samples

    if mnest:
        files = glob.glob('{}*'.format(basename))
        for f in files:
            os.remove(f)