Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
jitingxu1 committed Sep 12, 2024
1 parent b1a1250 commit 6858224
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions tests/test_init.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
import importlib
import sys
import logging
from importlib import import_module, reload
from unittest import mock

import pytest
import importlib


@pytest.mark.parametrize("optional_dep", ["pandas", "numpy", "pyarrow"])
def test_optional_dependencies(optional_dep, caplog):
Expand All @@ -16,9 +17,10 @@ def test_optional_dependencies(optional_dep, caplog):

assert "ibis_ml" in sys.modules


def test_import():
try:
mod = importlib.import_module("ibis_ml")
importlib.import_module("ibis_ml")
except AttributeError:
raise ImportError(f"cannot import IbisML") from None
raise ImportError("cannot import IbisML") from None
assert "ibis_ml" in sys.modules

0 comments on commit 6858224

Please sign in to comment.