fix codestyle and test

.gitignore

@@ -8,7 +8,7 @@ dist/*
 *.iml
 .idea
 .coverage
-.ipynb_checkpoints
+*/.ipynb_checkpoints
 examples/state.png
 .tox
 .python-version

tests/test_core.py

@@ -431,20 +431,30 @@ def test_function_callbacks(self):
         self.assertTrue(m.after_state_change[0].called)
 
     def test_state_callbacks(self):
+
         class Model:
-            def on_enter_A(self): pass
-            def on_exit_A(self): pass
-            def on_enter_B(self): pass
-            def on_exit_B(self): pass
-        states = [
-                  State(name='A', on_enter='on_enter_A', on_exit='on_exit_A'),
-                  State(name='B', on_enter='on_enter_B', on_exit='on_exit_B')
-                  ]
+            def on_enter_A(self):
+                pass
+
+            def on_exit_A(self):
+                pass
+
+            def on_enter_B(self):
+                pass
+
+            def on_exit_B(self):
+                pass
+
+        states = [State(name='A', on_enter='on_enter_A', on_exit='on_exit_A'),
+                  State(name='B', on_enter='on_enter_B', on_exit='on_exit_B')]
+
         machine = Machine(Model(), states=states)
-        self.assertEqual(len(states[0].on_enter), 1)
-        self.assertEqual(len(states[0].on_enter), 1)
-        self.assertEqual(len(states[1].on_exit), 1)
-        self.assertEqual(len(states[1].on_exit), 1)
+        state_a = machine.get_state('A')
+        state_b = machine.get_state('B')
+        self.assertEqual(len(state_a.on_enter), 1)
+        self.assertEqual(len(state_a.on_exit), 1)
+        self.assertEqual(len(state_b.on_enter), 1)
+        self.assertEqual(len(state_b.on_exit), 1)
 
     def test_pickle(self):
         import sys