fix tests

aidh-ms · Jan 7, 2025 · aedbe8c · aedbe8c
1 parent 6992967
commit aedbe8c
Show file tree

Hide file tree

Showing 10 changed files with 27 additions and 23 deletions.
diff --git a/tests/preprocessors/test_creatinie_preprocessor.py b/tests/preprocessors/test_creatinie_preprocessor.py
@@ -1,9 +1,10 @@
 from unittest import TestCase
-import pandas as pd
+
 import numpy as np
+import pandas as pd
 
-from pyAKI.preprocessors import CreatininePreProcessor
-from pyAKI.utils import Dataset, DatasetType
+from pyaki.preprocessors import CreatininePreProcessor
+from pyaki.utils import Dataset, DatasetType
 
 
 class TestCreatininePreProcessor(TestCase):

diff --git a/tests/preprocessors/test_demographics_preprocessor.py b/tests/preprocessors/test_demographics_preprocessor.py
@@ -1,8 +1,9 @@
 from unittest import TestCase
+
 import pandas as pd
 
-from pyAKI.preprocessors import DemographicsPreProcessor
-from pyAKI.utils import Dataset, DatasetType
+from pyaki.preprocessors import DemographicsPreProcessor
+from pyaki.utils import Dataset, DatasetType
 
 
 class TestDemographicsPreProcessor(TestCase):

diff --git a/tests/preprocessors/test_rrt_preprocessor.py b/tests/preprocessors/test_rrt_preprocessor.py
@@ -1,10 +1,10 @@
 from unittest import TestCase
-import pandas as pd
-import numpy as np
 
+import numpy as np
+import pandas as pd
 
-from pyAKI.preprocessors import RRTPreProcessor
-from pyAKI.utils import Dataset, DatasetType
+from pyaki.preprocessors import RRTPreProcessor
+from pyaki.utils import Dataset, DatasetType
 
 
 class TestRRTPreProcessor(TestCase):

diff --git a/tests/preprocessors/test_time_index_preprocessor.py b/tests/preprocessors/test_time_index_preprocessor.py
@@ -1,8 +1,9 @@
 from unittest import TestCase
+
 import pandas as pd
 
-from pyAKI.preprocessors import TimeIndexCreator
-from pyAKI.utils import Dataset, DatasetType
+from pyaki.preprocessors import TimeIndexCreator
+from pyaki.utils import Dataset, DatasetType
 
 
 class TestTimeIndexCreator(TestCase):

diff --git a/tests/preprocessors/test_urineoutput_preprocessor.py b/tests/preprocessors/test_urineoutput_preprocessor.py
@@ -1,9 +1,10 @@
 from unittest import TestCase
-import pandas as pd
+
 import numpy as np
+import pandas as pd
 
-from pyAKI.preprocessors import UrineOutputPreProcessor
-from pyAKI.utils import Dataset, DatasetType
+from pyaki.preprocessors import UrineOutputPreProcessor
+from pyaki.utils import Dataset, DatasetType
 
 
 class TestUrineOutputPreProcessor(TestCase):

diff --git a/tests/probes/test_creatinine_probes.py b/tests/probes/test_creatinine_probes.py
@@ -3,8 +3,8 @@
 import numpy as np
 import pandas as pd
 
-from pyAKI.kdigo import Analyser
-from pyAKI.probes import (
+from pyaki.kdigo import Analyser
+from pyaki.probes import (
     AbsoluteCreatinineProbe,
     AbstractCreatinineProbe,
     CreatinineBaselineMethod,

diff --git a/tests/probes/test_rrt_probe.py b/tests/probes/test_rrt_probe.py
@@ -2,8 +2,8 @@
 
 import pandas as pd
 
-from pyAKI.kdigo import Analyser
-from pyAKI.probes import Dataset, DatasetType, RRTProbe
+from pyaki.kdigo import Analyser
+from pyaki.probes import Dataset, DatasetType, RRTProbe
 from tests.set_up import setup_validation_data
 
 

diff --git a/tests/probes/test_urineoutput_probes.py b/tests/probes/test_urineoutput_probes.py
@@ -3,8 +3,8 @@
 import numpy as np
 import pandas as pd
 
-from pyAKI.kdigo import Analyser
-from pyAKI.probes import Dataset, DatasetType, UrineOutputMethod, UrineOutputProbe
+from pyaki.kdigo import Analyser
+from pyaki.probes import Dataset, DatasetType, UrineOutputMethod, UrineOutputProbe
 from tests.set_up import setup_validation_data
 
 

diff --git a/tests/set_up.py b/tests/set_up.py
@@ -2,7 +2,7 @@
 
 
 def setup_validation_data():
-    validation_data = pd.read_csv("test/data/validation_data.csv")
+    validation_data = pd.read_csv("tests/data/validation_data.csv")
     # coerce stay_id to int
     validation_data["stay_id"] = validation_data["stay_id"].astype(int)
     # convert charttime to datetime

diff --git a/tests/test_analyser.py b/tests/test_analyser.py
@@ -2,8 +2,8 @@
 
 import pandas as pd
 
-from pyAKI.kdigo import Analyser
-from pyAKI.probes import Dataset, DatasetType
+from pyaki.kdigo import Analyser
+from pyaki.probes import Dataset, DatasetType
 from tests.set_up import setup_validation_data