From 4d9efa010b4135bc972ea4eaa930b47d80c1e4ba Mon Sep 17 00:00:00 2001 From: lpin0002 Date: Wed, 4 Sep 2024 11:13:24 +1000 Subject: [PATCH] test 1 units --- tests/test_1_utils.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_1_utils.py b/tests/test_1_utils.py index 42012dbaf..f625a87a4 100644 --- a/tests/test_1_utils.py +++ b/tests/test_1_utils.py @@ -1,5 +1,5 @@ from gammabayes import update_with_defaults, haversine - +from astropy import units as u def test_update_with_defaults(): test_default = {'a':1, 'b':2, 'c':3, 'd':4} @@ -10,8 +10,8 @@ def test_update_with_defaults(): assert test_target == {'a':[1,2,3,4], 'b':2, 'c':'This is a test ya c', 'd':4} def test_haversine(): - lon1, lat1, lon2, lat2 = 0,0,0,0 - assert haversine(lon1, lat1, lon2, lat2) == 0 + lon1, lat1, lon2, lat2 = 0*u.deg,0*u.deg,0*u.deg,0*u.deg + assert haversine(lon1, lat1, lon2, lat2).value == 0 - lon1, lat1, lon2, lat2 = 180,0,0,0 - assert haversine(lon1, lat1, lon2, lat2) == 180 \ No newline at end of file + lon1, lat1, lon2, lat2 = 180*u.deg,0*u.deg,0*u.deg,0*u.deg + assert haversine(lon1, lat1, lon2, lat2).value == 180 \ No newline at end of file