diff --git a/aiida_dataframe/cli.py b/aiida_dataframe/cli.py index e73b044..10277ec 100644 --- a/aiida_dataframe/cli.py +++ b/aiida_dataframe/cli.py @@ -5,6 +5,7 @@ directly into the 'verdi' command by using AiiDA-specific entry points like "aiida.cmdline.data" (both in the pyproject.toml file). """ + import sys import tempfile diff --git a/aiida_dataframe/data/__init__.py b/aiida_dataframe/data/__init__.py index 9060437..b758256 100644 --- a/aiida_dataframe/data/__init__.py +++ b/aiida_dataframe/data/__init__.py @@ -3,6 +3,7 @@ Register data types via the "aiida.data" entry point in pyproject.toml. """ + from .dataframe import PandasFrameData __all__ = ("PandasFrameData",) diff --git a/aiida_dataframe/data/dataframe.py b/aiida_dataframe/data/dataframe.py index e4409d9..97fc5d5 100644 --- a/aiida_dataframe/data/dataframe.py +++ b/aiida_dataframe/data/dataframe.py @@ -2,6 +2,7 @@ This module defines a AiiDA Data plugin for pandas DataFrames to be stored in the file repository as HDF5 files """ + from __future__ import annotations import hashlib diff --git a/conftest.py b/conftest.py index 25a081b..b1e7701 100644 --- a/conftest.py +++ b/conftest.py @@ -1,4 +1,5 @@ """pytest fixtures for simplified testing.""" + import pytest pytest_plugins = ["aiida.manage.tests.pytest_fixtures", "pytest_regressions"] diff --git a/docs/source/conf.py b/docs/source/conf.py index 1c56e9d..320c0df 100755 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -222,9 +222,9 @@ def run_apidoc(_): # See https://stackoverflow.com/a/30144019 env = os.environ.copy() - env[ - "SPHINX_APIDOC_OPTIONS" - ] = "members,special-members,private-members,undoc-members,show-inheritance" + env["SPHINX_APIDOC_OPTIONS"] = ( + "members,special-members,private-members,undoc-members,show-inheritance" + ) subprocess.check_call([cmd_path] + options, env=env) diff --git a/tests/__init__.py b/tests/__init__.py index 94cbb42..484b467 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -1,8 +1,9 @@ -""" Tests for the plugin. +"""Tests for the plugin. Includes both tests written in unittest style (test_cli.py) and tests written in pytest style (test_calculations.py). """ + import os TEST_DIR = os.path.dirname(os.path.realpath(__file__)) diff --git a/tests/test_cli.py b/tests/test_cli.py index f75f043..278c6ae 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -1,4 +1,5 @@ -""" Tests for command line interface.""" +"""Tests for command line interface.""" + from click.testing import CliRunner import numpy as np import pandas as pd diff --git a/tests/test_data.py b/tests/test_data.py index 1d9209c..bc58ea1 100644 --- a/tests/test_data.py +++ b/tests/test_data.py @@ -1,4 +1,4 @@ -""" Tests for Data plugin.""" +"""Tests for Data plugin.""" import numpy as np import pandas as pd