Annotate stack test

icecube · Sep 27, 2023 · 9bbc147 · 9bbc147
1 parent 0c97b81
commit 9bbc147
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_stack.py b/test/test_stack.py
@@ -18,7 +18,7 @@ def setUp(self):
             str(self.testdata / "test_spline_1d.fits")
         )
 
-    def testStack(self):
+    def testStack(self) -> None:
         spline = self.spline
         stack = photospline.SplineTable.stack([spline] * 3, [-1, 0, 1], stackOrder=2)
         self.assertEqual(spline.ndim, 1)
@@ -29,7 +29,7 @@ def testStack(self):
             stack.knots[1], [-3.6, -2.6, -1.6, -0.6, 0.4, 1.4, 2.4, 3.4]
         )
 
-    def testBadArgs(self):
+    def testBadArgs(self) -> None:
         with self.assertRaises(ValueError):
             photospline.SplineTable.stack([], [])
         with self.assertRaises(TypeError):