def test_init_from_python_with_add(self):
# create the list of Fisher matrices:
fisher_list = []
matrix = np.identity(3)
for j in range(3):
for i in range(3):
matrix[i, i] = i + j + 1
param_names = ['pa' + str(i + j) for i in range(3)]
param_names_latex = ['m' + str(i) for i in range(3)]
fiducial = [float(i) for i in range(3)]
fisher = fm.fisher_matrix(fisher_matrix=matrix,
param_names=param_names,
param_names_latex=param_names_latex,
fiducial=fiducial)
fisher.name = 'fisher' + str(j + 1)
fisher_list.append(fisher)
fisher_list_test = fpa.CosmicFish_FisherAnalysis()
fisher_list_test.add_fisher_matrix(fisher_list)
# quality check:
assert fisher_list_test.fisher_name_list == [
fish.name for fish in fisher_list_test.fisher_list
]
assert fisher_list_test.fisher_name_list == [
fish.name for fish in fisher_list
]
def setup(self):
# create a list of fisher matrices:
fisher_1 = fm.fisher_matrix(file_name=test_input +
'/dummy_fisher_matrix.dat')
fisher_1.name = 'realistic fisher'
self.fisher_list_test = fpa.CosmicFish_FisherAnalysis(
fisher_list=fisher_1)
def test_init_from_path_search_true(self):
# create the list of Fisher matrices:
fisher_list_test = fpa.CosmicFish_FisherAnalysis(
fisher_path=test_input, search_fisher_guess=True)
# quality check:
assert fisher_list_test.fisher_name_list == [
fish.name for fish in fisher_list_test.fisher_list
]
def test_init_search(self):
# create the two list with different methods:
fisher_list_test_1 = fpa.CosmicFish_FisherAnalysis(
fisher_path=test_input, search_fisher_guess=False)
fisher_list_test_2 = fpa.CosmicFish_FisherAnalysis(
fisher_path=test_input, search_fisher_guess=True)
fisher_list_test_3 = fpa.CosmicFish_FisherAnalysis(
fisher_path=[
test_input + '/dummy_fisher_matrix.dat',
test_input + '/dummy_fisher_matrix_2.dat',
test_input + '/dummy_fisher_matrix_derived.dat'
],
search_fisher_guess=True)
# check wether they catch the same fishers:
for fish_1 in fisher_list_test_1.fisher_list:
assert fish_1 in fisher_list_test_2.fisher_list
for fish_2 in fisher_list_test_2.fisher_list:
assert fish_2 in fisher_list_test_3.fisher_list
def test_init_with_add_raises(self):
# init the list:
fisher_list_test = fpa.CosmicFish_FisherAnalysis()
# check first raise:
assert_raises(ValueError, fisher_list_test.add_fisher_matrix, [0.0])
# check second raise:
matrix = np.identity(3)
for i in range(3):
matrix[i, i] = i + 1
fisher = fm.fisher_matrix(fisher_matrix=matrix)
# add a fisher without a name:
fisher_list_test.add_fisher_matrix(fisher)
# add another fisher with the same (empty) name:
assert_raises(ValueError, fisher_list_test.add_fisher_matrix, fisher)
def setup(self):
# create a list of fisher matrices:
fisher_list = []
matrix = np.identity(3)
for j in range(3):
for i in range(3):
matrix[i, i] = i + j + 1
param_names_latex = ['m' + str(i) for i in range(3)]
fiducial = [float(i) for i in range(3)]
fisher = fm.fisher_matrix(fisher_matrix=matrix,
param_names_latex=param_names_latex,
fiducial=fiducial)
fisher.name = 'fisher' + str(j + 1)
fisher_list.append(fisher)
self.fisher_list_test = fpa.CosmicFish_FisherAnalysis(
fisher_list=fisher_list)
def setup(self):
# create a list of fisher matrices:
fisher_list = []
num_param = 4
num_fish = 3
matrix = np.identity(num_param)
for j in xrange(num_fish):
for i in xrange(num_param):
#matrix[i,i] = i+j+1
matrix[i,i] = (i+1)/float(j+1)
param_names_latex = [ 'm'+str(i) for i in xrange(num_param) ]
fiducial = [ float(i) for i in xrange(num_param) ]
fisher = fm.fisher_matrix( fisher_matrix=matrix, param_names_latex=param_names_latex, fiducial=fiducial )
fisher.name = 'fisher'+str(j+1)
fisher_list.append( fisher )
self.fisher_list_test = fpa.CosmicFish_FisherAnalysis(fisher_list=fisher_list)
outroot = args.outroot
else:
outroot = os.path.join(os.path.splitext(files[0])[0])
params = args.params
output_format = args.format
if output_format is None:
output_format = 'pdf'
# warning for the ini file:
if args.ini_file is not None:
raise ValueError('Not yet implemented')
# import Fisher matrices:
if args.derived:
fishers = fpa.CosmicFish_FisherAnalysis(fisher_path=files,
with_derived=True)
else:
fishers = fpa.CosmicFish_FisherAnalysis(fisher_path=files,
with_derived=False)
# manipulating Fisher:
if args.sum_fish is not None:
fishers_temp = fpa.CosmicFish_FisherAnalysis()
fisher_list = fishers.get_fisher_matrix()
for fish in fisher_list[1:]:
fisher_list[0] = fisher_list[0] + fish
fisher_list[0].name = args.sum_fish
fishers_temp.add_fisher_matrix(fisher_list[0])
fishers = fishers_temp
if args.eliminate:
else:
files = []
#for si in range(summax):
# if sum_fish[si][0]:
# print("entering extend")
# files.extend(sum_fish[si])
print 'No single fishers to plot, using only the combined ones'
print(fishnames)
print("files")
print(files)
#MOD: too much putput here!
summing = range(summax)
if derived is not False:
fishers = fpa.CosmicFish_FisherAnalysis(fisher_path=files,
with_derived=True)
for si in range(summax):
if sum_fish[si][0]:
print 'NOT HERE'
summing[si] = fpa.CosmicFish_FisherAnalysis(
fisher_path=sum_fish[si], with_derived=True)
else:
fishers = fpa.CosmicFish_FisherAnalysis(fisher_path=files,
with_derived=False)
for si in range(summax):
if sum_fish[si][0]:
print("enter sum pair")
summing[si] = fpa.CosmicFish_FisherAnalysis(
fisher_path=sum_fish[si], with_derived=False)
fishers_temp = fpa.CosmicFish_FisherAnalysis()
