apply PR feedback

ethereum · Apr 18, 2019 · 0faeb2d · 0faeb2d
1 parent af18ca9
commit 0faeb2d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/eth2/beacon/chains/testnet/__init__.py b/eth2/beacon/chains/testnet/__init__.py
@@ -7,13 +7,12 @@
 from eth2.beacon.state_machines.forks.xiao_long_bao import (
     XiaoLongBaoStateMachine,
 )
-from .constants import (
-    TESTNET_CHAIN_ID,
-)
 from eth2.beacon.state_machines.forks.serenity.configs import (
     SERENITY_CONFIG,
 )
-
+from .constants import (
+    TESTNET_CHAIN_ID,
+)
 
 if TYPE_CHECKING:
     from eth2.beacon.typing import (  # noqa: F401

diff --git a/tests/eth2/beacon/state_machines/forks/test_fork_classes.py b/tests/eth2/beacon/state_machines/forks/test_fork_classes.py
@@ -1,10 +1,17 @@
 import pytest
-from eth2.beacon.state_machines.forks.xiao_long_bao import XiaoLongBaoStateMachine
+
+from eth2.beacon.state_machines.forks.serenity import (
+    SerenityStateMachine,
+)
+from eth2.beacon.state_machines.forks.xiao_long_bao import (
+    XiaoLongBaoStateMachine,
+)
 
 
 @pytest.mark.parametrize(
     "sm_klass",
     (
+        SerenityStateMachine,
         XiaoLongBaoStateMachine,
     )
 )