def testThrow(self):
g = AllGenerator((m1, m2), (), {})
try:
g.throw(NameError("A"))
self.fail()
except NameError, e:
self.assertEquals("A", str(e))
def testSendError(self):
g = AllGenerator((), (), {})
try:
g.send("Value")
self.fail()
except TypeError, e:
self.assertEquals("can't send non-None value to a just-started generator", str(e))
def testSendValueNotAllowed(self):
g = AllGenerator((m1,m2), (), {})
g.next()
try:
g.send("s1")
self.fail()
except AssertionError, e:
self.assertEquals("%s returned 's1'" % m1, str(e))
def testSendError(self):
g = AllGenerator((), (), {})
try:
g.send("Value")
self.fail()
except TypeError, e:
self.assertEquals(
"can't send non-None value to a just-started generator",
str(e))
def testDeclineMessage(self):
g = AllGenerator((m1, m2), (), {})
g.next()
r = g.throw(DeclineMessage) # effectively skips one result
self.assertEquals("m2", r)
g = AllGenerator((m1, m3, m2), (), {})
r = list(g)
self.assertEquals(["m1", "m2"], r)
def testCreateAllGeneratorWithWrongArgs(self):
try:
AllGenerator()
self.fail()
except TypeError, e:
self.assertEquals("Required argument 'methods' (pos 1) not found",
str(e))
def testDeclineMessage(self):
g = AllGenerator((m1, m2), (), {})
g.next()
r = g.throw(DeclineMessage) # effectively skips one result
self.assertEquals("m2", r)
g = AllGenerator((m1, m3, m2), (), {})
r = list(g)
self.assertEquals(["m1", "m2"], r)
def testAllGeneratorWithCompose(self):
from weightless.core import compose
def f1():
yield 3
def f2():
yield f1()
g = AllGenerator((f2, ), (), {})
c = compose(g)
self.assertEquals([3], list(c))
def testSendValueNotAllowed(self):
g = AllGenerator((m1, m2), (), {})
g.next()
try:
g.send("s1")
self.fail()
except AssertionError, e:
self.assertEquals("%s returned 's1'" % m1, str(e))
class Observable_C_Test(TestCase):
def testImportAllGenerator(self):
from weightless.core.ext import AllGenerator
self.assertTrue(AllGenerator is not None)
def testCreateAllGeneratorWithWrongArgs(self):
try:
AllGenerator()
self.fail()
except TypeError, e:
self.assertEquals("Required argument 'methods' (pos 1) not found",
str(e))
try:
AllGenerator({})
self.fail()
except TypeError, e:
self.assertEquals("__new__() argument 1 must be tuple, not dict",
str(e))
def testTwoMethods(self):
g = AllGenerator((m1, m2), (), {})
r = list(g)
self.assertEquals(["m1", "m2"], r)
def testOneMethod(self):
g = AllGenerator((m1, ), (), {})
r = list(g)
self.assertEquals(["m1"], r)
def testEmpty(self):
g = AllGenerator((), (), {})
r = list(g)
self.assertEquals([], r)
def testAllGeneratorIsCallable(self):
g = AllGenerator((m1, ), (), {})
r = g()
self.assertEquals("m1", r)
def testArgs(self):
g = AllGenerator((m4, m5), (1, ), {"a": "A"})
r = list(g)
self.assertEquals([1, "A"], r)
def testThrow(self):
g = AllGenerator((m1, m2), (), {})
try: g.throw(NameError("A")); self.fail()
except NameError, e:
self.assertEquals("A", str(e))
def testCreateAllGenerator(self):
g = AllGenerator((), (), {})
self.assertTrue(g)
def all(self, *args, **kwargs):
return AllGenerator(self.methods, args, kwargs)
try:
AllGenerator()
self.fail()
except TypeError, e:
self.assertEquals("Required argument 'methods' (pos 1) not found",
str(e))
try:
AllGenerator({})
self.fail()
except TypeError, e:
self.assertEquals("__new__() argument 1 must be tuple, not dict",
str(e))
try:
AllGenerator(())
self.fail()
except TypeError, e:
self.assertEquals("Required argument 'args' (pos 2) not found",
str(e))
try:
AllGenerator((), {})
self.fail()
except TypeError, e:
self.assertEquals("__new__() argument 2 must be tuple, not dict",
str(e))
try:
AllGenerator((), ())
self.fail()
