def test_iad_refitting(): """ Check that refitting the IAD gives posteriors that approximately match the official Hipparcos values. Only run the MCMC for a few steps because this is a unit test. """ post, myHipLogProb = nielsen_iad_refitting_test( '{}/HIP027321.d'.format(DATADIR), burn_steps=10, mcmc_steps=200, saveplot=None) # check that we get reasonable values for the posteriors of the refit IAD # (we're only running the MCMC for a few steps, so these are not strict) assert np.isclose(0, np.median(post[:, -1]), atol=0.1) assert np.isclose(myHipLogProb.plx0, np.median(post[:, 0]), atol=0.1)
# The Hipparcos ID of your target. Available on Simbad. hip_num = '078092' # Name/path for the plot this function will make saveplot = 'HD142527_IADrefit.png' # Location of the Hipparcos IAD file. IAD_file = '{}H{}.d'.format('./', hip_num) start = datetime.now() # run the fit print('Go get a coffee. This will take a few mins! :)') nielsen_iad_refitting_test(IAD_file, hip_num=hip_num, saveplot=saveplot, burn_steps=hip_burn_steps, mcmc_steps=hip_mcmc_steps) end = datetime.now() duration_mins = (end - start).total_seconds() / 60 print( "Done! This fit took {:.1f} mins on my machine.".format(duration_mins)) ### ACTUAL ORBITIZE ### data_table = read_input.read_file(filename) print(data_table) # system parameters