Update cosmo_unit_tests.sh