coverage

rmjarvis · Jun 23, 2022 · 96af63f · 96af63f
1 parent 3627daf
commit 96af63f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_input.py b/tests/test_input.py
@@ -657,6 +657,13 @@ def test_flag_select():
     with np.testing.assert_raises(RuntimeError):
         piff.Select.process(config['select'], stars1)
 
+    # Raises at different place if all stars are rejected.
+    config['select']['use_flag'] = 1
+    config['select']['reject_where'] = 'True'
+    with np.testing.assert_raises(RuntimeError):
+        piff.Select.process(config['select'], stars1)
+    del config['select']['reject_where']
+
     # Base class selectStars function is not implemented.
     select = piff.Select(config['select'])
     with np.testing.assert_raises(NotImplementedError):