MAINT: Lint things better

test/test_environment_bench.py

@@ -1,7 +1,9 @@
 import subprocess
 import os
-import pytest
 import json
+
+import pytest
+
 from . import tools
 
 ENVIRONMENTS = []

test/tools.py

@@ -16,6 +16,7 @@
 import subprocess
 import platform
 import http.server
+import importlib
 from os.path import abspath, join, dirname, relpath, isdir
 from contextlib import contextmanager
 from hashlib import sha256
@@ -95,10 +96,10 @@ def _check_conda():
 def _check_mamba():
  conda = _find_conda()
  try:
- # Attempt to import libmambapy to check if libmambapy is available
- import libmambapy
- subprocess.check_call([conda, 'build', '--version'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
- # If both checks pass, return True indicating Mamba is available and conda-build is installed
+ importlib.import_module('libmambapy')
+ subprocess.check_call([conda, 'build', '--version'],
+  stdout=subprocess.PIPE,
+  stderr=subprocess.PIPE)
  return True
  except (ImportError, subprocess.CalledProcessError, FileNotFoundError):
  return False