Merge pull request #2 from GeoMop/nodianosh-pytest

napsl funce pro generovani distence a pridal unit testynapsane na pytest

Author: IvaNodyanosh

…gn_source_fields/assign_source_fields.py …thesis/functions/assign_source_fields.pyfield_synthesis/assign_source_fields/assign_source_fields.py renamed to field_synthesis/functions/assign_source_fields.py field_synthesis/assign_source_fields/assign_source_fields.py renamed to field_synthesis/functions/assign_source_fields.py

@@ -0,0 +1,19 @@
+import math
+
+def calc_distance(count_points: int, area_size: float, free_space_ratio: float = 0.4) -> float:
+    """
+    Odhadne optimální minimální vzdálenost bodů ve 2D prostoru 
+    s ohledem na procento požadovaného volného prostoru.
+    """
+    if count_points <= 0 or area_size <= 0:
+        return 0.0
+        
+    total_area = area_size ** 2
+    occupied_ratio = 1.0 - free_space_ratio
+    occupied_area = total_area * occupied_ratio
+    
+    area_per_point = occupied_area / count_points
+    
+    min_distance = math.sqrt(area_per_point)
+    
+    return min_distance

field_synthesis/functions/calc_distance.py

@@ -1,7 +1,7 @@
 from scipy.stats.qmc import PoissonDisk
 import numpy as np
 
-def generate_anchor_points(count_points, min_distance, area_size):
+def generate_points(count_points, min_distance, area_size):
     """
     Funkce vygeneruje pole nahodnych bodu.
     

…_anchor_points/generate_anchor_points.py …d_synthesis/functions/generate_points.pyfield_synthesis/generate_anchor_points/generate_anchor_points.py renamed to field_synthesis/functions/generate_points.py field_synthesis/generate_anchor_points/generate_anchor_points.py renamed to field_synthesis/functions/generate_points.py

@@ -0,0 +1,28 @@
+import pytest
+import numpy as np
+from field_synthesis.functions.assign_source_fields import assign_source_fields
+
+def test_assign_source_fields_valid():
+    """Testuje správné hranice náhodných indexů a délku výstupu."""
+    anchor_points = np.array([[1, 1], [2, 2], [3, 3], [4, 4]])
+    num_fields = 3
+    
+    result = assign_source_fields(anchor_points, num_fields)
+    
+    assert len(result) == len(anchor_points)
+    assert np.issubdtype(result.dtype, np.integer)
+    # Indexy musí být 0, 1 nebo 2 (menší než num_fields)
+    assert np.all(result >= 0)
+    assert np.all(result < num_fields)
+
+@pytest.mark.parametrize("anchors, fields", [
+    (np.array([]), 5),        # Prázdné pole bodů
+    (np.array([[1, 1]]), 0),  # Nula zdrojových polí
+    (np.array([[1, 1]]), -1)  # Záporný počet polí
+])
+def test_assign_source_fields_invalid(anchors, fields):
+    """Testuje okrajové případy a neplatné vstupy."""
+    result = assign_source_fields(anchors, fields)
+    
+    assert isinstance(result, np.ndarray)
+    assert len(result) == 0

field_synthesis/generate_anchor_points/test_generate_anchor_points.py

@@ -0,0 +1,24 @@
+import pytest
+import math
+from field_synthesis.functions.calc_distance import calc_distance
+
+def test_estimate_distance_standard():
+    """Testuje standardní výpočet se 40% volným prostorem."""
+    # 50 bodů, plocha 100x100 (10000), 60% obsazeno = 6000. 6000/50 = 120. Odmocnina z 120 je cca 10.954
+    result = calc_distance(50, 100.0, 0.4)
+    assert math.isclose(result, 10.95445, rel_tol=1e-4)
+
+@pytest.mark.parametrize("points, area, free_space, expected", [
+    (10, 50.0, 0.5, 11.1803),
+    (100, 10.0, 0.1, 0.9486)
+])
+def test_estimate_distance_parametrized(points, area, free_space, expected):
+    """Testuje různé poměry a velikosti."""
+    result = calc_distance(points, area, free_space)
+    assert math.isclose(result, expected, rel_tol=1e-4)
+
+def test_estimate_distance_invalid_inputs():
+    """Testuje chování při neplatných vstupech (nula nebo záporná čísla)."""
+    assert calc_distance(0, 100.0, 0.4) == 0.0
+    assert calc_distance(10, 0.0, 0.4) == 0.0
+    assert calc_distance(-5, 100.0, 0.4) == 0.0

field_synthesis/tests/test_assign_source_fields.py

@@ -0,0 +1,30 @@
+import numpy as np
+from field_synthesis.functions.generate_points import generate_points
+
+def test_generate_points_shape():
+    """Testuje, zda funkce vrací 2D pole se správným tvarem."""
+    points = generate_points(count_points=20, min_distance=2.0, area_size=50.0)
+    
+    assert isinstance(points, np.ndarray)
+    assert points.shape[1] == 2  # Musí mít x a y souřadnice
+    assert len(points) <= 20     # PoissonDisk nemusí vždy vygenerovat maximum bodů
+
+def test_generate_points_invalid_inputs():
+    """Testuje pojistky proti nulovým a záporným hodnotám."""
+    # Očekáváme prázdné 2D pole: tvar (0, 2)
+    empty_result = np.zeros((0, 2))
+    
+    result_zero_points = generate_points(0, 5.0, 50.0)
+    result_zero_area = generate_points(10, 5.0, 0.0)
+    
+    np.testing.assert_array_equal(result_zero_points, empty_result)
+    np.testing.assert_array_equal(result_zero_area, empty_result)
+
+def test_generate_points_bounds():
+    """Testuje, zda všechny vygenerované body leží uvnitř zadané oblasti."""
+    area = 50.0
+    points = generate_points(10, 2.0, area)
+    
+    # Žádná souřadnice nesmí být menší než 0 nebo větší než area_size
+    assert np.all(points >= 0.0)
+    assert np.all(points <= area)

field_synthesis/tests/test_distance.py

@@ -1,2 +1,3 @@
-numpy
-scipy
+numpy>=2.0.0
+scipy>=1.14.0
+pytest