Applies PEP-8 recommendations.

Whitespace wrangling.
This commit is contained in:
Eddie Hebert
2012-08-25 04:10:06 -04:00
parent 06742e27a1
commit 6f195233f5
+5 -8
View File
@@ -13,23 +13,21 @@ from zipline.utils.test_utils import create_receiver, drain_receiver
from unittest2 import TestCase
class LoggerTestCase(TestCase):
def setUp(self):
configure_logging()
self.LOG = logging.getLogger("ZiplineLogger")
configure_logging()
self.LOG = logging.getLogger("ZiplineLogger")
def test_log(self):
test_msg = uuid.uuid1().hex
self.LOG.info(test_msg)
logfile = open('/var/log/zipline/zipline.log','r')
logfile = open('/var/log/zipline/zipline.log', 'r')
with logfile:
last_line = tail(logfile, window=1)
logged_msg = last_line.split(" - ")[1]
self.assertEqual(test_msg, logged_msg)
def test_zmq_handler(self):
socket_addr = 'tcp://127.0.0.1:10000'
ctx = zmq.Context()
@@ -37,8 +35,8 @@ class LoggerTestCase(TestCase):
socket_push.connect(socket_addr)
recv = create_receiver(socket_addr, ctx)
zmq_out = ZeroMQLogHandler(
socket = socket_push,
filter = lambda r, h: r.channel in ['test zmq logger'],
socket=socket_push,
filter=lambda r, h: r.channel in ['test zmq logger'],
context=ctx,
#bubble=False
)
@@ -50,7 +48,6 @@ class LoggerTestCase(TestCase):
with zmq_out.threadbound():
log.info(ex.num)
output, _ = drain_receiver(recv, count=1)
self.assertEqual(output[-1]['prefix'], 'LOG')
self.assertTrue(isinstance(output[-1]['payload']['msg'], basestring))