From 3cfe84039b19885eff3fe2427dac3016f83d4e48 Mon Sep 17 00:00:00 2001 From: MARCHAND MANON Date: Fri, 20 Oct 2023 14:48:54 +0200 Subject: [PATCH] test: add try/except for DALOverflowWarning --- astroquery/simbad/tests/test_simbad_remote.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/astroquery/simbad/tests/test_simbad_remote.py b/astroquery/simbad/tests/test_simbad_remote.py index 8670d477d9..861a3c6590 100644 --- a/astroquery/simbad/tests/test_simbad_remote.py +++ b/astroquery/simbad/tests/test_simbad_remote.py @@ -11,7 +11,11 @@ # Maybe we need to expose SimbadVOTableResult to be in the public API? from astroquery.simbad.core import SimbadVOTableResult from astroquery.exceptions import BlankResponseWarning -from pyvo.dal import DALOverflowWarning +try: + # This requires pyvo 1.4.2 + from pyvo.dal.exceptions import DALOverflowWarning +except ImportError: + pass # M42 coordinates @@ -267,6 +271,7 @@ def test_query_tap(self): expect = "letters numbers\n------- -------\n a 1\n b 2\n c 3" assert expect == str(result) # Test query_tap raised errors + # DALOverflowWarning exists since pyvo 1.4.2 with pytest.raises(DALOverflowWarning, match="Partial result set *"): truncated_result = Simbad.query_tap("SELECT * from basic", maxrec=2) assert len(truncated_result) == 2