pim/pg-rad
1
0
Fork 0
mirror of https://github.com/pim-n/pg-rad synced 2026-02-02 14:33:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

move test_objects.py to test_sources.py and update to reflect refactor

Browse Source
This commit is contained in:
Pim Nelissen
2026-01-28 13:50:16 +01:00
parent 85f306a469
commit caec70b39b
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_objects.py → tests/test_sources.py
View File

@ -1,28 +1,27 @@
import numpy as np import numpy as np
import pytest import pytest
from pg_rad.objects import Source from pg_rad.sources import PointSource
@pytest.fixture @pytest.fixture
def test_sources(): def test_sources():
pos_a = np.random.rand(3) pos_a = np.random.rand(3)
pos_b = np.random.rand(3) pos_b = np.random.rand(3)
a = Source(*tuple(pos_a), strength = None) a = PointSource(*tuple(pos_a), strength = None)
b = Source(*tuple(pos_b), strength = None) b = PointSource(*tuple(pos_b), strength = None)
return pos_a, pos_b, a, b return pos_a, pos_b, a, b
def test_if_distances_equal(test_sources): def test_if_distances_equal(test_sources):
"""_Verify whether from object A to object B is the same as B to A._""" """Verify whether from PointSource A to PointSource B is the same as B to A."""
_, _, a, b = test_sources _, _, a, b = test_sources
assert a.distance_to(b) == b.distance_to(a) assert a.distance_to(b) == b.distance_to(a)
def test_distance_calculation(test_sources): def test_distance_calculation(test_sources):
"""_Verify whether distance between two static objects (e.g. sources) """Verify whether distance between two PointSources is calculated correctly."""
is calculated correctly._"""
pos_a, pos_b, a, b = test_sources pos_a, pos_b, a, b = test_sources