aboutsummaryrefslogtreecommitdiff
path: root/test/iterators.tm
blob: 0b6c2a8968bf9391afed66ffff03f0bf838df520 (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 none: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 none: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