Test current cosmology module

The current cosmology module is relatively simple, some tests to show its accuracy to distance calculations, and further updates to the cosmology module to allow for more than FlatLambdaCDM would be welcome while we wait for phytorch