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
|
# mode: run
# tag: cpp, werror, cpp11
from cython.operator cimport dereference as deref
from libcpp cimport bool
from libcpp.algorithm cimport merge, inplace_merge
from libcpp.vector cimport vector
cdef bool less(int a, int b):
return a < b
def test_merge(vector[int] v1, vector[int] v2):
"""
Test merge.
>>> test_merge([1, 3, 5], [2, 4])
[1, 2, 3, 4, 5]
"""
cdef vector[int] out = vector[int](v1.size() + v2.size())
merge(v1.begin(), v1.end(), v2.begin(), v2.end(), out.begin())
return out
def test_merge_with_bin_pred(vector[int] v1, vector[int] v2):
"""
Test merge with binary predicate
>>> test_merge_with_bin_pred([1, 3, 5], [2, 4])
[1, 2, 3, 4, 5]
"""
cdef vector[int] out = vector[int](v1.size() + v2.size())
merge(v1.begin(), v1.end(), v2.begin(), v2.end(), out.begin(), less)
return out
def test_inplace_merge(vector[int] v):
"""
Test inplace_merge.
>>> test_inplace_merge([4, 5, 6, 1, 2, 3])
[1, 2, 3, 4, 5, 6]
"""
inplace_merge(v.begin(), v.begin() + 3, v.end())
return v
def test_inplace_merge_with_bin_pred(vector[int] v):
"""
Test inplace_merge with binary predicate
>>> test_inplace_merge_with_bin_pred([4, 5, 6, 1, 2, 3])
[1, 2, 3, 4, 5, 6]
"""
inplace_merge(v.begin(), v.begin() + 3, v.end(), less)
return v
|
