Develop

tests/test_trends/test_interface.py

@@ -18,6 +18,14 @@ def test_calculate_trend_wrong_quantile(mean_data):
         trend = calculate_trend("quant", mean_data, 1.2)
 
 
+def test_calculate_trend_wrong_num_samples(mean_data):
+    with pytest.raises(
+            ValueError,
+            match="The number of samples must be a positive integer."
+    ):
+        trend = calculate_trend("quant", mean_data, 0.5, -1)
+
+
 @pytest.mark.filterwarnings("ignore:Maximum number of iterations")
 @pytest.mark.slow
 def test_calculate_trend_one_quantile(mean_data):
@@ -25,6 +33,12 @@ def test_calculate_trend_one_quantile(mean_data):
     assert list(trend.keys()) == [0.5]
 
 
+@pytest.mark.filterwarnings("ignore:Maximum number of iterations")
+def test_calculate_trend_one_quantile_few_samples(mean_data):
+    trend = calculate_trend("quant", mean_data, 0.5, 50)
+    assert list(trend.keys()) == [0.5]
+
+
 @pytest.mark.filterwarnings("ignore:Maximum number of iterations")
 @pytest.mark.slow
 def test_calculate_trend_multiple_quantiles(mean_data):
@@ -32,7 +46,18 @@ def test_calculate_trend_multiple_quantiles(mean_data):
     assert list(trends.keys()) == [0.25, 0.5, 0.75]
 
 
+@pytest.mark.filterwarnings("ignore:Maximum number of iterations")
+def test_calculate_trend_multiple_quantiles_few_samples(mean_data):
+    trends = calculate_trend("quant", mean_data, [0.25, 0.5, 0.75], 50)
+    assert list(trends.keys()) == [0.25, 0.5, 0.75]
+
+
 @pytest.mark.slow
 def test_calculate_trend_ols(mean_data):
     trend = calculate_trend("OLS", mean_data)
     assert list(trend.keys()) == ["trend", "uncertainty", "p_value"]
+
+
+def test_calculate_trend_ols_few_samples(mean_data):
+    trend = calculate_trend("OLS", mean_data, num_samples=50)
+    assert list(trend.keys()) == ["trend", "uncertainty", "p_value"]