Style fixes.

andes/system.py

@@ -17,9 +17,10 @@
 import os
 import sys
 import inspect
-import dill
 import time
 
+import dill
+
 from collections import OrderedDict
 from typing import Dict, Tuple, Union, Optional
 
@@ -37,9 +38,8 @@
 from andes.io.streaming import Streaming
 
 from andes.shared import np, jac_names, dilled_vars
-from andes.shared import matrix, spmatrix, sparse, Pool
+from andes.shared import matrix, spmatrix, sparse, Pool, Process
 
-from multiprocess import Process
 
 logger = logging.getLogger(__name__)
 dill.settings['recurse'] = True

andes/utils/__init__.py

@@ -1,4 +1,4 @@
 from andes.utils import paths           # NOQA
 from andes.utils import sympy           # NOQA
 from andes.utils.paths import get_case  # NOQA
-from andes.utils import snapshot        # NOQA
+from andes.utils import snapshot        # NOQA

tests/test_snapshot.py

@@ -1,8 +1,12 @@
+"""
+Test ANDES snapshot based on dill.
+"""
+
 import os
-import andes
+import unittest
 import numpy as np
 
-import unittest
+import andes
 from andes.utils.snapshot import save_ss, load_ss
 
 
@@ -12,6 +16,10 @@ class TestSnapshot(unittest.TestCase):
     """
 
     def test_save_ss(self):
+        """
+        Test saving a snapshot.
+        """
+
         ss = andes.run(andes.get_case("kundur/kundur_full.xlsx"))
         ss.TDS.config.tf = 2
         ss.TDS.run()
@@ -20,6 +28,9 @@ def test_save_ss(self):
         os.remove('test_ss.pkl')
 
     def load_ss(self):
+        """
+        Test loading a snapshot and continuing the simulation.
+        """
 
         # load a snapshot
         test_dir = os.path.dirname(__file__)