Fixed PEP8

Sergio Oliveira
1 parent 1b82e3cc
Showing 1 changed file with 26 additions and 31 deletions Show diff stats
colab/signals/tests/test_signals.py
@@ -3,50 +3,48 @@ Test Signals class.
 Objective: Test parameters, and behavior.
 """
-from django.test import TestCase
-from django.apps import apps
 import django
-from colab.signals.signals import *
+
+from django.test import TestCase
+
 from mock import patch, MagicMock, PropertyMock
+from colab.signals.signals import (registered_signals, register_signal,
+                                   connect_signal, send)
+
 class SignalsTest(TestCase):
     def setUp(self):
-       django.setup()
-       self.list_signal = ['a', 'b', 'c']
-       self.plugin_name = 'test_signal'
-
+        django.setup()
+        self.list_signal = ['a', 'b', 'c']
+        self.plugin_name = 'test_signal'
     def test_register_signal_(self):
-       register_signal(self.plugin_name, self.list_signal)
-
-       signal_name ='a'
-       signal_list = ['test_signal']
-       self.assertEqual(len(registered_signals[signal_name]), 1)
-       self.assertEqual(registered_signals[signal_name], signal_list)
+        register_signal(self.plugin_name, self.list_signal)
+        signal_name = 'a'
+        signal_list = ['test_signal']
+        self.assertEqual(len(registered_signals[signal_name]), 1)
+        self.assertEqual(registered_signals[signal_name], signal_list)
-   
     def test_register_signal_already_registered(self):
-       signal_name ='a'
-       signal_list = ['test_signal']
+        signal_name = 'a'
+        signal_list = ['test_signal']
-       register_signal(self.plugin_name, self.list_signal)
-       self.assertEqual(len(registered_signals[signal_name]), 1)
+        register_signal(self.plugin_name, self.list_signal)
+        self.assertEqual(len(registered_signals[signal_name]), 1)
-       register_signal(self.plugin_name, self.list_signal)
-       self.assertEqual(len(registered_signals[signal_name]), 1)
-       self.assertEqual(registered_signals[signal_name], signal_list)
+        register_signal(self.plugin_name, self.list_signal)
+        self.assertEqual(len(registered_signals[signal_name]), 1)
+        self.assertEqual(registered_signals[signal_name], signal_list)
-    
     def test_connect_non_registered_signal(self):
         sender = 'Test'
         handling_method = 'Test'
         signal_name = 'Test'
-        
-        self.assertRaises(Exception, connect_signal, signal_name,
-                sender, handling_method) 
+        self.assertRaises(Exception, connect_signal, signal_name,
+                          sender, handling_method)
     @patch('colab.signals.signals.Signal.connect')
     def test_connect_already_registered_signal(self, mock):
@@ -54,23 +52,21 @@ class SignalsTest(TestCase):
         handling_method = MagicMock
         type(handling_method).delay = PropertyMock(return_value='Test')
         signal_name = 'a'
-        
+
         register_signal(self.plugin_name, self.list_signal)
         connect_signal(signal_name, sender, handling_method.delay)
         args, kwargs = mock.call_args
-        
+
         self.assertEqual(args[0], handling_method)
         self.assertEqual(kwargs['sender'], sender)
         self.assertTrue(mock.is_called)
-        
     @patch('colab.signals.signals.Signal.send')
     def test_send_signal(self, mock):
         sender = 'Test'
-        handling_method = 'Test'
         signal_name = 'a'
-        
+
         register_signal(self.plugin_name, self.list_signal)
         send(signal_name, sender)
@@ -78,7 +74,6 @@ class SignalsTest(TestCase):
         self.assertEqual(kwargs['sender'], sender)
         self.assertTrue(mock.is_called)
-        
     def test_send_signal_not_registered(self):
         self.assertRaises(Exception, send, 'test_signal', 'test')