Clean up imports

JuBiotech · Oct 11, 2024 · 72e012c · 72e012c
1 parent 38a8875
commit 72e012c
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 25 deletions.
diff --git a/bletl/core.py b/bletl/core.py
@@ -1,16 +1,8 @@
 """Specifies the base types for parsing and representing BioLector CSV files."""
-import abc
-import enum
-import os
-import pathlib
-import typing
-import urllib.error
-import urllib.request
-import warnings
 from collections.abc import Iterable
+from pathlib import Path
 from typing import Optional, Sequence, Union
 
-import numpy
 import pandas
 
 from . import parsing, utils
@@ -20,11 +12,6 @@
     BLDParser,
     FilterTimeSeries,
     IncompatibleFileError,
-    InvalidLotNumberError,
-    LotInformationError,
-    LotInformationMismatch,
-    LotInformationNotFound,
-    NoMeasurementData,
 )
 
 parsers = {
@@ -33,12 +20,12 @@
 }
 
 
-def get_parser(filepath: Union[str, pathlib.Path]) -> BLDParser:
+def get_parser(filepath: Union[str, Path]) -> BLDParser:
     """Analyzes a raw BioLector file and selects an appropiate parser.
 
     Parameters
     ----------
-    filepath : str or pathlib.Path
+    filepath : str or Path
         Path pointing to the file of interest.
 
     Returns
@@ -90,7 +77,7 @@ def get_parser(filepath: Union[str, pathlib.Path]) -> BLDParser:
 
 
 def _parse(
-    filepath: str,
+    filepath: Union[str, Path],
     drop_incomplete_cycles: bool,
     lot_number: Optional[int],
     temp: Optional[int],
@@ -105,7 +92,7 @@ def _parse(
 
     Parameters
     ----------
-    filepath : str or pathlib.Path
+    filepath : str or Path
         Path pointing to the file of interest.
     drop_incomplete_cycles : bool
         If `True`, incomplete cycles at the end are discarded.
@@ -160,7 +147,7 @@ def _parse(
 
 
 def parse(
-    filepaths: Union[str, Sequence[str]],
+    filepaths: Union[Union[str, Path], Sequence[Union[str, Path]]],
     *,
     drop_incomplete_cycles: bool = True,
     lot_number: Optional[int] = None,
@@ -176,7 +163,7 @@ def parse(
 
     Parameters
     ----------
-    filepaths : str or pathlib.Path or iterable
+    filepaths : str or Path or iterable
         Path pointing to the file(s) of interest.
         If an iterable is provided, files are concatenated.
     drop_incomplete_cycles : bool

diff --git a/bletl/utils.py b/bletl/utils.py
@@ -5,6 +5,8 @@
 import pathlib
 import re
 import urllib
+import urllib.error
+import urllib.request
 from typing import Optional, Sequence, Tuple
 
 import pandas

diff --git a/tests/test_core.py b/tests/test_core.py
@@ -5,10 +5,16 @@
 import numpy
 import pandas
 import pytest
-from numpy import random
 
 import bletl
-from bletl import core, parsing, utils
+from bletl import (
+    InvalidLotNumberError,
+    LotInformationMismatch,
+    NoMeasurementData,
+    core,
+    parsing,
+    utils,
+)
 from bletl.parsing import bl1, blpro
 
 dir_testfiles = pathlib.Path(pathlib.Path(__file__).absolute().parent, "data")
@@ -302,7 +308,7 @@ def test_get_unified_narrow_data(self):
             data.get_unified_narrow_data(source_well="O9000")
 
     def test_NoMeasurements_Warning(self):
-        with pytest.warns(core.NoMeasurementData):
+        with pytest.warns(NoMeasurementData):
             bletl.parse(file_with_no_measurements)
 
 
@@ -369,7 +375,7 @@ def test_fragments_with_cal_data(self):
         return
 
     def test_mismatch_warning(self):
-        with pytest.warns(core.LotInformationMismatch):
+        with pytest.warns(LotInformationMismatch):
             bletl.parse(file_with_lot_info, lot_number=1818, temp=37)
         return
 
@@ -381,7 +387,7 @@ def test_get_calibration_dict(self):
         return
 
     def test_invalid_lot_number(self):
-        with pytest.raises(core.InvalidLotNumberError):
+        with pytest.raises(InvalidLotNumberError):
             bl1.fetch_calibration_data(99, 99)
         return