aboutsummaryrefslogtreecommitdiff
path: root/test/iterators.tm
blob: fca91b2dec7418d332550af2054dc114473bdd35 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

struct Pair(x:Text, y:Text)

func pairwise(strs:[Text] -> func(->Pair?)):
    i := 1
    return func():
        if i + 1 > strs.length: return !Pair
        i += 1
        return Pair(strs[i-1], strs[i])?

func range(first:Int, last:Int -> func(->Int?)):
    i := first
    return func():
        if i > last:
            return !Int
        i += 1
        return (i-1)?

func main():
    values := ["A", "B", "C", "D"]

    >> ((++) "($(foo.x)$(foo.y))" for foo in pairwise(values))
    = "(AB)(BC)(CD)"
    >> ["$(foo.x)$(foo.y)" for foo in pairwise(values)]
    = ["AB", "BC", "CD"]

    do:
        result := [:Text]
        for foo in pairwise(values):
            result:insert("$(foo.x)$(foo.y)")
        >> result
        = ["AB", "BC", "CD"]

    >> [i for i in range(5, 10)]
    = [5, 6, 7, 8, 9, 10]

    >> (+) range(5, 10)
    = 45