make list iterator in plottests

NNPDF · Mar 11, 2024 · 616d4e7 · 616d4e7
1 parent e84533b
commit 616d4e7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/validphys2/src/validphys/tests/test_plots.py b/validphys2/src/validphys/tests/test_plots.py
@@ -18,7 +18,7 @@ def test_plotpdfs():
     Q = 10
     flavours = ['g']
     # plot_pdfs returns a generator with (figure, name_hint)
-    return API.plot_pdfs(pdfs=pdfs, Q=Q, flavours=flavours)[0]
+    return next(iter(API.plot_pdfs(pdfs=pdfs, Q=Q, flavours=flavours)))[0]
 
 
 @pytest.mark.linux
@@ -44,13 +44,13 @@ def test_dataspecschi2():
 @pytest.mark.linux
 @pytest.mark.mpl_image_compare(tolerance=TOLERANCE_VALUE)
 def test_plot_smpdf(single_data_internal_cuts_config):
-    return API.plot_smpdf(**single_data_internal_cuts_config)
+    return next(iter(API.plot_smpdf(**single_data_internal_cuts_config)))
 
 
 @pytest.mark.linux
 @pytest.mark.mpl_image_compare(tolerance=TOLERANCE_VALUE)
 def test_plot_smpdf_categorical(single_data_categorical_internal_cuts_config):
-    return API.plot_smpdf(**single_data_categorical_internal_cuts_config)
+    return next(iter(API.plot_smpdf(**single_data_categorical_internal_cuts_config)))
 
 
 @pytest.mark.linux