From 72cc44266ef583d47f7da283cab4537a318bcc64 Mon Sep 17 00:00:00 2001 From: Sherelyn Alejandro <80668767+SherelynA@users.noreply.github.com> Date: Mon, 4 Nov 2024 17:13:16 -0500 Subject: [PATCH] added numpy import --- sedkit/tests/test_isochrone.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/sedkit/tests/test_isochrone.py b/sedkit/tests/test_isochrone.py index 039c089..fad39f4 100644 --- a/sedkit/tests/test_isochrone.py +++ b/sedkit/tests/test_isochrone.py @@ -2,18 +2,19 @@ import unittest import pytest import astropy.units as q +import numpy as np from .. import isochrone as iso from .. import utilities as u @pytest.mark.parametrize('xval,age,xparam,yparam,expected_result', [ - ((-4, 0.1), (4 * q.Gyr, 0.1 * q.Gyr), 'Lbol', 'mass',0.072), # With uncertainties - (-4, (4 * q.Gyr, 0.1 * q.Gyr), 'Lbol', 'mass',0.072), # No xparam uncertainty - ((-4, 0.1), 4 * q.Gyr, 'Lbol', 'mass',0.072), # No yparam uncertainty - (-4, 4 * q.Gyr, 'Lbol', 'mass',0.020) # No xparam and yparam uncertainties + ((-4, 0.1), (4 * q.Gyr, 0.1 * q.Gyr), 'Lbol', 'mass', 0.072), # With uncertainties + (-4, (4 * q.Gyr, 0.1 * q.Gyr), 'Lbol', 'mass', 0.072), # No xparam uncertainty + ((-4, 0.1), 4 * q.Gyr, 'Lbol', 'mass', 0.072), # No yparam uncertainty + (-4, 4 * q.Gyr, 'Lbol', 'mass', 0.020) # No xparam and yparam uncertainties ]) -def test_evaluate( xval, age, xparam, yparam,expected_result): +def test_evaluate(xval, age, xparam, yparam, expected_result): # average, lower, upper """Test the evaluate method""" hsa = iso.Isochrone('hybrid_solar_age')