bugfix: sample_between failing on small ranges

yasserfarouk · Dec 25, 2023 · dddd1eb · dddd1eb
1 parent e18eb18
commit dddd1eb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/negmas/preferences/generators.py b/negmas/preferences/generators.py
@@ -55,6 +55,8 @@ def sample_between(
     """
     if n == 0:
         return []
+    if end - start < 1e-8:
+        return [start] * n
     samples = np.linspace(start, end, num=n, endpoint=endpoint)
     if len(samples) > 2:
         samples[1:-1] += (

diff --git a/tests/core/test_ufun_generators.py b/tests/core/test_ufun_generators.py
@@ -62,9 +62,11 @@ def test_distribute_integer_randomly_on_none(n, m):
     n=st.integers(1, 100),
     endpoint=st.booleans(),
     main_range_min=st.floats(0.0, 0.5),
-    main_range_range=st.floats(0, 0.5),
+    main_range_range=st.floats(0.0, 0.5),
 )
 def test_sample_between(start, rng, n, endpoint, main_range_min, main_range_range):
+    main_range_min = round(main_range_min, 6)
+    main_range_range = round(main_range_range, 6)
     end = start + rng
     lst = sample_between(
         start, end, n, endpoint, (main_range_min, main_range_min + main_range_range)