2021-06-05 22:47:59 +08:00
|
|
|
|
|
|
|
def f0()
|
|
|
|
yield("yield value")
|
|
|
|
end
|
|
|
|
|
2021-06-23 12:18:15 +08:00
|
|
|
import Fiber
|
|
|
|
|
|
|
|
fiber = Fiber.new(f0)
|
|
|
|
yield_value = Fiber.run(fiber)
|
2021-06-05 22:47:59 +08:00
|
|
|
assert(yield_value == "yield value")
|
2021-06-23 12:18:15 +08:00
|
|
|
assert(!fiber.is_done)
|
2021-06-05 22:47:59 +08:00
|
|
|
|
|
|
|
def f1()
|
|
|
|
assert(yield("y") == "r")
|
|
|
|
yield()
|
|
|
|
end
|
2021-06-23 12:18:15 +08:00
|
|
|
fiber = Fiber.new(f1)
|
|
|
|
assert(Fiber.run(fiber) == "y")
|
|
|
|
Fiber.resume(fiber, "r")
|
|
|
|
assert(!fiber.is_done)
|
2021-06-05 22:47:59 +08:00
|
|
|
|
|
|
|
def f2(p1, p2, p3)
|
|
|
|
r1 = yield(p3); assert(r1 == 'r1')
|
|
|
|
r2 = yield(p2); assert(r2 == 'r2')
|
|
|
|
r3 = yield(p1); assert(r3 == 'r3')
|
|
|
|
|
|
|
|
return p1 + p2 * p3
|
|
|
|
end
|
|
|
|
|
2021-06-23 12:18:15 +08:00
|
|
|
fiber = Fiber.new(f2)
|
|
|
|
p3 = Fiber.run(fiber, 1, 2, 3); assert(p3 == 3)
|
|
|
|
p2 = Fiber.resume(fiber, 'r1'); assert(p2 == 2)
|
|
|
|
p1 = Fiber.resume(fiber, 'r2'); assert(p1 == 1)
|
|
|
|
pa = Fiber.resume(fiber, 'r3'); assert(pa == 7)
|
|
|
|
assert(fiber.is_done)
|
2021-06-16 02:54:30 +08:00
|
|
|
|
|
|
|
# If we got here, that means all test were passed.
|
|
|
|
print('All TESTS PASSED')
|