about summary refs log tree commit diff
path: root/third_party/immer/example/flex-vector/flex-vector.cpp
blob: 7c8a7793393cb094cf0fca87aa4f4b85c71da0c4 (plain) (blame)
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
//
// immer: immutable data structures for C++
// Copyright (C) 2016, 2017, 2018 Juan Pedro Bolivar Puente
//
// This software is distributed under the Boost Software License, Version 1.0.
// See accompanying file LICENSE or copy at http://boost.org/LICENSE_1_0.txt
//

#include <immer/flex_vector.hpp>

#include <cassert>

int main()
{
    {
        // include:push-back/start
        auto v1 = immer::flex_vector<int>{1};
        auto v2 = v1.push_back(8);

        assert((v1 == immer::flex_vector<int>{1}));
        assert((v2 == immer::flex_vector<int>{1, 8}));
        // include:push-back/end
    }
    {
        // include:push-front/start
        auto v1 = immer::flex_vector<int>{1};
        auto v2 = v1.push_front(8);

        assert((v1 == immer::flex_vector<int>{1}));
        assert((v2 == immer::flex_vector<int>{8, 1}));
        // include:push-front/end
    }

    {
        // include:set/start
        auto v1 = immer::flex_vector<int>{1, 2, 3};
        auto v2 = v1.set(0, 5);

        assert((v1 == immer::flex_vector<int>{1, 2, 3}));
        assert((v2 == immer::flex_vector<int>{5, 2, 3}));
        // include:set/end
    }

    {
        // include:update/start
        auto v1 = immer::flex_vector<int>{1, 2, 3, 4};
        auto v2 = v1.update(2, [&](auto l) { return ++l; });

        assert((v1 == immer::flex_vector<int>{1, 2, 3, 4}));
        assert((v2 == immer::flex_vector<int>{1, 2, 4, 4}));
        // include:update/end
    }

    {
        // include:take/start
        auto v1 = immer::flex_vector<int>{1, 2, 3, 4, 5, 6};
        auto v2 = v1.take(3);

        assert((v1 == immer::flex_vector<int>{1, 2, 3, 4, 5, 6}));
        assert((v2 == immer::flex_vector<int>{1, 2, 3}));
        // include:take/end
    }

    {
        // include:drop/start
        auto v1 = immer::flex_vector<int>{1, 2, 3, 4, 5, 6};
        auto v2 = v1.drop(3);

        assert((v1 == immer::flex_vector<int>{1, 2, 3, 4, 5, 6}));
        assert((v2 == immer::flex_vector<int>{4, 5, 6}));
        // include:drop/end
    }

    {
        // include:insert/start
        auto v1 = immer::flex_vector<int>{1, 2, 3};
        auto v2 = v1.insert(0, 0);

        assert((v1 == immer::flex_vector<int>{1, 2, 3}));
        assert((v2 == immer::flex_vector<int>{0, 1, 2, 3}));
        // include:insert/end
    }

    {
        // include:erase/start
        auto v1 = immer::flex_vector<int>{1, 2, 3, 4, 5};
        auto v2 = v1.erase(2);

        assert((v1 == immer::flex_vector<int>{1, 2, 3, 4, 5}));
        assert((v2 == immer::flex_vector<int>{1, 2, 4, 5}));
        // include:erase/end
    }

    {
        // include:concat/start
        auto v1 = immer::flex_vector<int>{1, 2, 3};
        auto v2 = v1 + v1;

        assert((v1 == immer::flex_vector<int>{1, 2, 3}));
        assert((v2 == immer::flex_vector<int>{1, 2, 3, 1, 2, 3}));
        // include:concat/end
    }
}