Skip to content

Commit ca4ddc1

Browse files
committed
Fix class instance mocking
1 parent c8c19c6 commit ca4ddc1

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

astroquery/esa/jwst/tests/test_jwsttap.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -926,6 +926,9 @@ def __check_extracted_files(self, files_expected, files_returned):
926926

927927
def test_query_target(self):
928928
jwst = JwstClass()
929+
simbad = Simbad()
930+
ned = Ned()
931+
vizier = Vizier()
929932
# Testing default parameters
930933
with pytest.raises(ValueError) as err:
931934
jwst.query_target(target_name="M1", target_resolver="")
@@ -942,13 +945,13 @@ def test_query_target(self):
942945
# Testing no valid coordinates from resolvers
943946
simbad_file = data_path('test_query_by_target_name_simbad_ned_error.vot')
944947
simbad_table = Table.read(simbad_file)
945-
Simbad.query_object = MagicMock(return_value=simbad_table)
948+
simbad.query_object = MagicMock(return_value=simbad_table)
946949
ned_file = data_path('test_query_by_target_name_simbad_ned_error.vot')
947950
ned_table = Table.read(ned_file)
948-
Ned.query_object = MagicMock(return_value=ned_table)
951+
ned.query_object = MagicMock(return_value=ned_table)
949952
vizier_file = data_path('test_query_by_target_name_vizier_error.vot')
950953
vizier_table = Table.read(vizier_file)
951-
Vizier.query_object = MagicMock(return_value=vizier_table)
954+
vizier.query_object = MagicMock(return_value=vizier_table)
952955

953956
coordinate_error = 'coordinate must be either a string or astropy.coordinates'
954957
with pytest.raises(ValueError) as err:
@@ -980,14 +983,14 @@ def test_query_target(self):
980983
parameters['upload_table_name'] = None
981984
simbad_file = data_path('test_query_by_target_name_simbad.vot')
982985
simbad_table = Table.read(simbad_file)
983-
Simbad.query_object = MagicMock(return_value=simbad_table)
986+
simbad.query_object = MagicMock(return_value=simbad_table)
984987
ned_file = data_path('test_query_by_target_name_ned.vot')
985988
ned_table = Table.read(ned_file)
986-
Ned.query_object = MagicMock(return_value=ned_table)
989+
ned.query_object = MagicMock(return_value=ned_table)
987990
vizier_file = data_path('test_query_by_target_name_vizier.vot')
988991
vizier_table = Table.read(vizier_file)
989992
vizier_table_list = TableList({'1': vizier_table})
990-
Vizier.query_object = MagicMock(return_value=vizier_table_list)
993+
vizier.query_object = MagicMock(return_value=vizier_table_list)
991994

992995
with open(data_path('test_query_by_target_name_simbad_query.txt'), 'r') as file:
993996
parameters['query'] = file.read()

0 commit comments

Comments
 (0)