benchmark_lambdacdm_test.go

package cosmo

import (
	"reflect"
	"testing"
)

func benchmarkLambdaCDMEN(n int, b *testing.B) {
	cos := LambdaCDM{H0: 70, Om0: 0.3, Ol0: 0.9}

	var z float64
	zMax := 1.0
	step := zMax / float64(n)
	for i := 0; i < b.N; i++ {
		for j := 0; j < n; j++ {
			z = 0.001 + step*float64(j)
			cos.E(z)
		}
	}
}

func BenchmarkLambdaCDMEN(b *testing.B) {
	benchmarkLambdaCDMEN(10000, b)
}

func BenchmarkLambdaCDMENdistance(b *testing.B) {
	benchmarkLambdaCDMNdistance(10000, "E", b)
}

func BenchmarkLambdaCDME(b *testing.B) {
	cos := LambdaCDM{H0: 70, Om0: 0.3, Ol0: 0.9}
	z := 1.0
	for i := 0; i < b.N; i++ {
		cos.E(z)
	}
}

func BenchmarkLambdaCDMEinv(b *testing.B) {
	cos := LambdaCDM{H0: 70, Om0: 0.3, Ol0: 0.9}
	z := 1.0
	for i := 0; i < b.N; i++ {
		cos.Einv(z)
	}
}

// benchmarkLambdaCDMDistanceFlat is a helper function to be called by specific benchmarkLambdaCDMs
//   for an Omega_K == 0 cosmology
func benchmarkLambdaCDMDistanceFlat(distFunc string, b *testing.B) {
	cos := LambdaCDM{H0: 70, Om0: 0.3, Ol0: 0.7}
	z := 1.0

	funcToTest := reflect.ValueOf(&cos).MethodByName(distFunc)
	for i := 0; i < b.N; i++ {
		funcToTest.Call([]reflect.Value{reflect.ValueOf(z)})
	}
}

// benchmarkLambdaCDMDistancePositiveOk0 is a helper function to be called by specific benchmarkLambdaCDMs
//   for an Omega_K > 0 cosmology
func benchmarkLambdaCDMDistancePositiveOk0(distFunc string, b *testing.B) {
	cos := LambdaCDM{H0: 70, Om0: 0.3, Ol0: 0.9}
	z := 1.0

	funcToTest := reflect.ValueOf(&cos).MethodByName(distFunc)
	for i := 0; i < b.N; i++ {
		funcToTest.Call([]reflect.Value{reflect.ValueOf(z)})
	}
}

// benchmarkLambdaCDMDistanceOM is a helper function to be called by specific benchmarkLambdaCDMs
//   for an Omega_Lambda = 0 cosmology
func benchmarkLambdaCDMDistanceOM(distFunc string, b *testing.B) {
	cos := LambdaCDM{H0: 70, Om0: 0.27, Ol0: 0.}
	z := 1.0

	funcToTest := reflect.ValueOf(&cos).MethodByName(distFunc)
	for i := 0; i < b.N; i++ {
		funcToTest.Call([]reflect.Value{reflect.ValueOf(z)})
	}
}

// benchmarkLambdaCDMDistance is a helper function to be called by specific benchmarkLambdaCDMs
func benchmarkLambdaCDMDistance(distFunc string, b *testing.B) {
	cos := LambdaCDM{H0: 70, Om0: 0.3, Ol0: 0.9}
	z := 1.0

	funcToTest := reflect.ValueOf(&cos).MethodByName(distFunc)
	for i := 0; i < b.N; i++ {
		funcToTest.Call([]reflect.Value{reflect.ValueOf(z)})
	}
}

// benchmarkLambdaCDMNdistance is a helper function to be called by specific benchmarkLambdaCDMs
func benchmarkLambdaCDMNdistance(n int, distFunc string, b *testing.B) {
	cos := LambdaCDM{H0: 70, Om0: 0.3, Ol0: 0.9}
	funcToTest := reflect.ValueOf(&cos).MethodByName(distFunc)
	var z float64
	zMax := 1.0
	step := zMax / float64(n)
	for i := 0; i < b.N; i++ {
		for j := 0; j < n; j++ {
			z = 0.001 + step*float64(j)
			funcToTest.Call([]reflect.Value{reflect.ValueOf(z)})
		}
	}
}

func BenchmarkLambdaCDMComovingDistance(b *testing.B) {
	benchmarkLambdaCDMDistance("ComovingDistance", b)
}

func BenchmarkLambdaCDMComovingTransverseDistance(b *testing.B) {
	benchmarkLambdaCDMDistance("ComovingTransverseDistance", b)
}

func BenchmarkLambdaCDMLuminosityDistance(b *testing.B) {
	benchmarkLambdaCDMDistance("LuminosityDistance", b)
}

func BenchmarkLambdaCDMLookbackTime(b *testing.B) {
	benchmarkLambdaCDMDistance("LookbackTime", b)
}

func BenchmarkLambdaCDMNComovingDistance(b *testing.B) {
	benchmarkLambdaCDMNdistance(10000, "ComovingDistance", b)
}

func BenchmarkLambdaCDMNLuminosityDistance(b *testing.B) {
	benchmarkLambdaCDMNdistance(10000, "LuminosityDistance", b)
}

func BenchmarkLambdaCDMNE(b *testing.B) {
	benchmarkLambdaCDMNdistance(10000, "E", b)
}

func BenchmarkLambdaCDMComovingDistanceOM(b *testing.B) {
	benchmarkLambdaCDMDistanceOM("ComovingDistance", b)
}

func BenchmarkLambdaCDMLookbackTimeOM(b *testing.B) {
	benchmarkLambdaCDMDistanceOM("LookbackTime", b)
}

func BenchmarkLambdaCDMComovingDistanceFlat(b *testing.B) {
	benchmarkLambdaCDMDistanceFlat("ComovingDistance", b)
}

func BenchmarkLambdaCDMComovingDistancePositiveOk0(b *testing.B) {
	benchmarkLambdaCDMDistancePositiveOk0("ComovingDistance", b)
}