Merge pull request #55 from MilesCranmer/deepsource-transform-e6f88263

Format code with black
MilesCranmer · Jun 8, 2021 · 072271b · 072271b
2 parents 5af6354 + 717bfae
commit 072271b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test.py b/test/test.py
@@ -149,9 +149,9 @@ def test_feature_selection_handler(self):
             np.sort(selected_X, axis=1), np.sort(X[:, [2, 3]], axis=1)
         )
 
-class TestHelperFunctions(unittest.TestCase):
 
-    @patch('builtins.input', side_effect=['y', 'n'])
+class TestHelperFunctions(unittest.TestCase):
+    @patch("builtins.input", side_effect=["y", "n"])
     def test_yesno(self, mock_input):
         # Assert that the yes/no function correctly deals with y/n
         self.assertEqual(_yesno("Test"), True)