|
1 #include <vector> |
|
2 #include <algorithm> |
|
3 #include <functional> |
|
4 #include <numeric> |
|
5 |
|
6 #include "iota.h" |
|
7 #include "cppunit/cppunit_proxy.h" |
|
8 |
|
9 #if !defined (STLPORT) || defined(_STLP_USE_NAMESPACES) |
|
10 using namespace std; |
|
11 #endif |
|
12 |
|
13 // |
|
14 // TestCase class |
|
15 // |
|
16 class MergeTest : public CPPUNIT_NS::TestCase |
|
17 { |
|
18 CPPUNIT_TEST_SUITE(MergeTest); |
|
19 CPPUNIT_TEST(merge0); |
|
20 CPPUNIT_TEST(merge1); |
|
21 CPPUNIT_TEST(merge2); |
|
22 CPPUNIT_TEST_SUITE_END(); |
|
23 |
|
24 protected: |
|
25 void merge0(); |
|
26 void merge1(); |
|
27 void merge2(); |
|
28 }; |
|
29 |
|
30 CPPUNIT_TEST_SUITE_REGISTRATION(MergeTest); |
|
31 |
|
32 // |
|
33 // tests implementation |
|
34 // |
|
35 void MergeTest::merge0() |
|
36 { |
|
37 int numbers1[5] = { 1, 6, 13, 25, 101 }; |
|
38 int numbers2[5] = {-5, 26, 36, 46, 99 }; |
|
39 |
|
40 int result[10]; |
|
41 merge((int*)numbers1, (int*)numbers1 + 5, (int*)numbers2, (int*)numbers2 + 5, (int*)result); |
|
42 |
|
43 CPPUNIT_ASSERT(result[0]==-5); |
|
44 CPPUNIT_ASSERT(result[1]==1); |
|
45 CPPUNIT_ASSERT(result[2]==6); |
|
46 CPPUNIT_ASSERT(result[3]==13); |
|
47 CPPUNIT_ASSERT(result[4]==25); |
|
48 CPPUNIT_ASSERT(result[5]==26); |
|
49 CPPUNIT_ASSERT(result[6]==36); |
|
50 CPPUNIT_ASSERT(result[7]==46); |
|
51 CPPUNIT_ASSERT(result[8]==99); |
|
52 CPPUNIT_ASSERT(result[9]==101); |
|
53 } |
|
54 void MergeTest::merge1() |
|
55 { |
|
56 vector<int> v1(5); |
|
57 vector<int> v2(v1.size()); |
|
58 __iota(v1.begin(), v1.end(), 0); |
|
59 __iota(v2.begin(), v2.end(), 3); |
|
60 |
|
61 vector <int> result(v1.size() + v2.size()); |
|
62 merge(v1.begin(), v1.end(), v2.begin(), v2.end(), result.begin()); |
|
63 |
|
64 CPPUNIT_ASSERT(result[0]==0); |
|
65 CPPUNIT_ASSERT(result[1]==1); |
|
66 CPPUNIT_ASSERT(result[2]==2); |
|
67 CPPUNIT_ASSERT(result[3]==3); |
|
68 CPPUNIT_ASSERT(result[4]==3); |
|
69 CPPUNIT_ASSERT(result[5]==4); |
|
70 CPPUNIT_ASSERT(result[6]==4); |
|
71 CPPUNIT_ASSERT(result[7]==5); |
|
72 CPPUNIT_ASSERT(result[8]==6); |
|
73 CPPUNIT_ASSERT(result[9]==7); |
|
74 |
|
75 } |
|
76 void MergeTest::merge2() |
|
77 { |
|
78 vector <int> v1(5); |
|
79 vector <int> v2(v1.size()); |
|
80 for (int i = 0; (size_t)i < v1.size(); ++i) { |
|
81 v1[i] = 10 - i; |
|
82 v2[i] = 7 - i; |
|
83 } |
|
84 vector<int> result(v1.size() + v2.size()); |
|
85 merge(v1.begin(), v1.end(), v2.begin(), v2.end(), result.begin(), greater<int>() ); |
|
86 |
|
87 CPPUNIT_ASSERT(result[0]==10); |
|
88 CPPUNIT_ASSERT(result[1]==9); |
|
89 CPPUNIT_ASSERT(result[2]==8); |
|
90 CPPUNIT_ASSERT(result[3]==7); |
|
91 CPPUNIT_ASSERT(result[4]==7); |
|
92 CPPUNIT_ASSERT(result[5]==6); |
|
93 CPPUNIT_ASSERT(result[6]==6); |
|
94 CPPUNIT_ASSERT(result[7]==5); |
|
95 CPPUNIT_ASSERT(result[8]==4); |
|
96 CPPUNIT_ASSERT(result[9]==3); |
|
97 } |