pocketlang/tests/lang/fibers.pk

39 lines
786 B
Plaintext
Raw Normal View History

2021-06-05 22:47:59 +08:00
def f0()
yield("yield value")
end
import Fiber
fiber = Fiber.new(f0)
yield_value = Fiber.run(fiber)
2021-06-05 22:47:59 +08:00
assert(yield_value == "yield value")
assert(!fiber.is_done)
2021-06-05 22:47:59 +08:00
def f1()
assert(yield("y") == "r")
yield()
end
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
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')