@@ -66,14 +66,17 @@ class ConversationHandlerTest(BaseTest, unittest.TestCase):
6666 def setUp (self ):
6767 self .current_state = dict ()
6868 self .entry_points = [CommandHandler ('start' , self .start )]
69- self .states = {self .THIRSTY : [CommandHandler ('brew' , self .brew ),
70- CommandHandler ('wait' , self .start )],
71- self .BREWING : [CommandHandler ('pourCoffee' , self .drink )],
72- self .DRINKING : [CommandHandler ('startCoding' , self .code ),
73- CommandHandler ('drinkMore' , self .drink )],
74- self .CODING : [CommandHandler ('keepCoding' , self .code ),
75- CommandHandler ('gettingThirsty' , self .start ),
76- CommandHandler ('drinkMore' , self .drink )],}
69+ self .states = {
70+ self .THIRSTY : [CommandHandler ('brew' , self .brew ), CommandHandler ('wait' , self .start )],
71+ self .BREWING : [CommandHandler ('pourCoffee' , self .drink )],
72+ self .DRINKING :
73+ [CommandHandler ('startCoding' , self .code ), CommandHandler ('drinkMore' , self .drink )],
74+ self .CODING : [
75+ CommandHandler ('keepCoding' , self .code ),
76+ CommandHandler ('gettingThirsty' , self .start ),
77+ CommandHandler ('drinkMore' , self .drink )
78+ ],
79+ }
7780 self .fallbacks = [CommandHandler ('eat' , self .start )]
7881
7982 def _setup_updater (self , * args , ** kwargs ):
0 commit comments