@@ -347,17 +347,14 @@ def test_votable_fields():
347
347
sb .add_votable_fields ('rot' , 'z_value' , 'velocity' )
348
348
assert (set (sb .get_votable_fields ())
349
349
== set (['main_id' , 'coordinates' , 'rot' , 'z_value' , 'velocity' ]))
350
- try :
351
- sb .add_votable_fields ('velocity' )
352
- except KeyError :
353
- pass # this is the expected response
350
+
354
351
assert (set (sb .get_votable_fields ())
355
352
== set (['main_id' , 'coordinates' , 'rot' , 'z_value' , 'velocity' ]))
356
353
sb .remove_votable_fields ('rot' , 'main_id' , 'coordinates' )
357
354
assert set (sb .get_votable_fields ()) == set (['z_value' , 'velocity' ])
358
355
# Warning is expected as we removed the 'coordinates' field above:
359
356
with pytest .warns (UserWarning , match = "coordinates: this field is not set" ):
360
- sb .remove_votable_fields ('rot' , 'main_id' , ' coordinates' )
357
+ sb .remove_votable_fields ('coordinates' )
361
358
assert set (sb .get_votable_fields ()) == set (['z_value' , 'velocity' ])
362
359
with pytest .warns (UserWarning , match = "All fields have been removed. Resetting" ):
363
360
sb .remove_votable_fields ('z_value' , 'velocity' )
0 commit comments