Somre more basic tests.
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…
Reference in New Issue