Added extra unit test

nanograv · Nov 9, 2023 · 4b94571 · 4b94571
1 parent 8b75650
commit 4b94571
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_pulsar.py b/tests/test_pulsar.py
@@ -255,6 +255,15 @@ def test_droppsr(self):
         with self.assertRaises(AttributeError):
             _ = self.psr_nodrop.timfile
 
+    def test_drop_not_picklable(self):
+        self.psr_nodrop.drop_not_picklable()
+
+        with self.assertRaises(AttributeError):
+            _ = self.psr_nodrop.model
+
+        with self.assertRaises(AttributeError):
+            _ = self.psr_nodrop.pint_toas
+
     def test_deflate_inflate(self):
         pass