Somre more basic tests.

pull/35/head
David Wilson 8 years ago
parent b964cc6f18
commit 4de557d304

@ -0,0 +1,23 @@
#!/usr/bin/env python
import socket
import time
import unittest
import econtext.master
import econtext.utils
@econtext.utils.with_broker
def do_stuff(broker):
context = econtext.master.connect(broker)
t0 = time.time()
ncalls = 1000
for x in xrange(ncalls):
context.call(socket.gethostname)
return (1e6 * (time.time() - t0)) / ncalls
class LocalContextTimingTest(unittest.TestCase):
def test_timing(self):
self.assertLess(do_stuff(), 1000)

@ -0,0 +1,32 @@
#!/usr/bin/env python
import unittest
import econtext.master
import econtext.utils
def func0(broker):
return broker
@econtext.utils.with_broker
def func(broker):
return broker
class RunWithBrokerTest(unittest.TestCase):
# test_shutdown_on_exception
# test_shutdown_on_success
def test_run_with_broker(self):
broker = econtext.utils.run_with_broker(func0)
self.assertTrue(isinstance(broker, econtext.master.Broker))
self.assertFalse(broker._thread.isAlive())
class WithBrokerTest(unittest.TestCase):
def test_with_broker(self):
broker = func()
self.assertTrue(isinstance(broker, econtext.master.Broker))
self.assertFalse(broker._thread.isAlive())
Loading…
Cancel
Save