Lint, fix imports

system/__init__.py

@@ -4,7 +4,6 @@
 """
 
 # Add imports here
-from .system import *
 
 # Handle versioneer
 from ._version import get_versions

system/system.py

@@ -1,4 +1,4 @@
-from typing import Union, Iterable, List, Dict
+from typing import Union, Iterable, Dict
 
 from pydantic import BaseModel, validator
 import pint

system/tests/test_potential.py

@@ -8,6 +8,7 @@
 
 u = pint.UnitRegistry()
 
+
 class TestPotential(BaseTest):
     def test_analytical_potential_constructor(self):
         pot = AnalyticalPotential(

system/tests/test_system.py

@@ -1,12 +1,12 @@
-import pytest
 import pint
 
-from system import System, PotentialHandler, PotentialCollection, Topology, handler_conversion
+from system import System, PotentialHandler, PotentialCollection, handler_conversion
 from system.tests.base_test import BaseTest
 
 
 u = pint.UnitRegistry()
 
+
 class TestPotentialHandler(BaseTest):
     def test_handler_conversion(self, argon_ff):
         collection = handler_conversion(
@@ -21,6 +21,7 @@ def test_handler_conversion(self, argon_ff):
 
         assert collection['vdW']['[#18:1]'].parameters['sigma'] == 0.3 * u.nm
 
+
 class TestSystem(BaseTest):
     def test_constructor(self, argon_ff, argon_top):
         """Test the basic constructor"""

system/tests/test_utils.py

@@ -1,5 +1,4 @@
 import pytest
-import sympy
 from simtk import unit as simtk_unit
 import pint
 
@@ -8,6 +7,7 @@
 
 u = pint.UnitRegistry()
 
+
 def test_simtk_to_pint():
     """Test conversion from SimTK Quantity to pint Quantity."""
     simtk_quantity = 10.0 * simtk_unit.nanometer
@@ -16,11 +16,13 @@ def test_simtk_to_pint():
 
     assert pint_quantity == 10.0 * u.nanometer
 
+
 def test_pint_to_simtk():
     """Test conversion from pint Quantity to SimTK Quantity."""
     with pytest.raises(NotImplementedError):
         pint_to_simtk(None)
 
+
 @pytest.mark.parametrize(
     'expr1,expr2,result',
     [
@@ -31,4 +33,4 @@ def test_pint_to_simtk():
     ]
 )
 def test_compare_sympy_expr(expr1, expr2, result):
-    compare_sympy_expr(expr1, expr2) == result
+    assert compare_sympy_expr(expr1, expr2) == result

system/typing/smirnoff.py

@@ -56,10 +56,12 @@ def handler_conversion(forcefield, potential_collection, handler_name):
 
     return potential_collection
 
+
 class SMIRNOFFPotentialHandler(PotentialHandler):
 
     pass
 
+
 class SMIRNOFFCollection(PotentialCollection):
 
     @classmethod

system/utils.py

@@ -34,6 +34,7 @@ def compare_sympy_expr(expr1, expr2):
 
     return sympy.simplify(expr1 - expr2) == 0
 
+
 def get_test_file_path(test_file):
     dir_path = resource_filename('system', 'tests/files/')
     test_file_path = pathlib.Path(dir_path).joinpath(test_file)