remove extra nonsense from Claude

Author: John Wu

tests/core/test_mimic3.py

@@ -11,26 +11,18 @@
 class TestMIMIC3Demo(unittest.TestCase):
     """Test MIMIC3 dataset with demo data downloaded from PhysioNet."""
 
-    # Class-level variables for shared dataset
-    temp_dir = None
-    demo_dataset_path = None
-    dataset = None
-
-    @classmethod
-    def setUpClass(cls):
-        """Download demo dataset once for all tests."""
-        cls.temp_dir = tempfile.mkdtemp()
-        cls._download_demo_dataset()
-        cls._load_dataset()
-
-    @classmethod
-    def tearDownClass(cls):
-        """Clean up downloaded dataset after all tests."""
-        if cls.temp_dir and os.path.exists(cls.temp_dir):
-            shutil.rmtree(cls.temp_dir)
-
-    @classmethod
-    def _download_demo_dataset(cls):
+    def setUp(self):
+        """Download and set up demo dataset for each test."""
+        self.temp_dir = tempfile.mkdtemp()
+        self._download_demo_dataset()
+        self._load_dataset()
+
+    def tearDown(self):
+        """Clean up downloaded dataset after each test."""
+        if self.temp_dir and os.path.exists(self.temp_dir):
+            shutil.rmtree(self.temp_dir)
+
+    def _download_demo_dataset(self):
         """Download MIMIC-III demo dataset using wget."""
         download_url = "https://physionet.org/files/mimiciii-demo/1.4/"
 
@@ -42,7 +34,7 @@ def _download_demo_dataset(cls):
             "-c",
             "-np",
             "--directory-prefix",
-            cls.temp_dir,
+            self.temp_dir,
             download_url,
         ]
 
@@ -55,19 +47,17 @@ def _download_demo_dataset(cls):
 
         # Find the downloaded dataset path
         physionet_dir = (
-            Path(cls.temp_dir) / "physionet.org" / "files" / "mimiciii-demo" / "1.4"
+            Path(self.temp_dir) / "physionet.org" / "files" / "mimiciii-demo" / "1.4"
         )
         if physionet_dir.exists():
-            cls.demo_dataset_path = str(physionet_dir)
+            self.demo_dataset_path = str(physionet_dir)
         else:
             raise unittest.SkipTest("Downloaded dataset not found in expected location")
 
-    @classmethod
-    def _load_dataset(cls):
-        """Load the dataset once for all tests."""
+    def _load_dataset(self):
+        """Load the dataset for testing."""
         tables = ["diagnoses_icd", "procedures_icd", "prescriptions", "noteevents"]
-
-        cls.dataset = MIMIC3Dataset(root=cls.demo_dataset_path, tables=tables)
+        self.dataset = MIMIC3Dataset(root=self.demo_dataset_path, tables=tables)
 
     def test_stats(self):
         """Test .stats() method execution."""