1 #include <numeric>
2 #include <string>
3 #include <iterator>
4 #include <vector>
5 #include <algorithm>
6 #include <functional>
7
8 #include "iota.h"
9 #include "cppunit/cppunit_proxy.h"
10
11 #if !defined (STLPORT) || defined(_STLP_USE_NAMESPACES)
12 using namespace std;
13 #endif
14
15 //
16 // TestCase class
17 //
18 class SetDifferenceTest : public CPPUNIT_NS::TestCase
19 {
20 CPPUNIT_TEST_SUITE(SetDifferenceTest);
21 CPPUNIT_TEST(setdiff0);
22 CPPUNIT_TEST(setdiff1);
23 CPPUNIT_TEST(setdiff2);
24 CPPUNIT_TEST(setsymd0);
25 CPPUNIT_TEST(setsymd1);
26 CPPUNIT_TEST(setsymd2);
27 CPPUNIT_TEST_SUITE_END();
28
29 protected:
30 void setdiff0();
31 void setdiff1();
32 void setdiff2();
33 void setsymd0();
34 void setsymd1();
35 void setsymd2();
36 };
37
38 CPPUNIT_TEST_SUITE_REGISTRATION(SetDifferenceTest);
39
40 //
41 // tests implementation
42 //
setsymd0()43 void SetDifferenceTest::setsymd0()
44 {
45 int v1[3] = { 13, 18, 23 };
46 int v2[4] = { 10, 13, 17, 23 };
47 int result[4] = { 0, 0, 0, 0 };
48
49 set_symmetric_difference((int*)v1, (int*)v1 + 3, (int*)v2, (int*)v2 + 4, (int*)result);
50 CPPUNIT_ASSERT(result[0]==10);
51 CPPUNIT_ASSERT(result[1]==17);
52 CPPUNIT_ASSERT(result[2]==18);
53 CPPUNIT_ASSERT(result[3]==0);
54 }
55
setsymd1()56 void SetDifferenceTest::setsymd1()
57 {
58 vector<int> v1(10);
59 __iota(v1.begin(), v1.end(), 0);
60 vector<int> v2(10);
61 __iota(v2.begin(), v2.end(), 7);
62
63 vector<int> diff;
64 set_symmetric_difference(v1.begin(), v1.end(), v2.begin(), v2.end(), back_inserter(diff));
65 CPPUNIT_ASSERT( diff.size() == 14 );
66 int int_res[] = {0, 1, 2, 3, 4, 5, 6, 10, 11, 12, 13, 14, 15, 16};
67 for (int i = 0; i < 14; ++i) {
68 CPPUNIT_ASSERT( diff[i] == int_res[i] );
69 }
70 }
71
setsymd2()72 void SetDifferenceTest::setsymd2()
73 {
74 const char* word1 = "ABCDEFGHIJKLMNO";
75 const char* word2 = "LMNOPQRSTUVWXYZ";
76
77 string diff;
78 set_symmetric_difference(word1, word1 + ::strlen(word1), word2, word2 + ::strlen(word2),
79 back_inserter(diff), less<char>());
80 CPPUNIT_ASSERT( diff.size() == 22 );
81 char char_res[] = "ABCDEFGHIJKPQRSTUVWXYZ";
82 for (int i = 0; i < 22; ++i) {
83 CPPUNIT_ASSERT( diff[i] == char_res[i] );
84 }
85 }
86
setdiff0()87 void SetDifferenceTest::setdiff0()
88 {
89 int v1[3] = { 13, 18, 23 };
90 int v2[4] = { 10, 13, 17, 23 };
91 int result[4] = { 0, 0, 0, 0 };
92 //18 0 0 0
93 //10 17 23 0
94
95 set_difference((int*)v1, (int*)v1 + 3, (int*)v2, (int*)v2 + 4, (int*)result);
96 CPPUNIT_ASSERT( result[0] == 18 );
97 CPPUNIT_ASSERT( result[1] == 0 );
98 CPPUNIT_ASSERT( result[2] == 0 );
99 CPPUNIT_ASSERT( result[3] == 0 );
100
101 set_difference((int*)v2, (int*)v2 + 4, (int*)v1, (int*)v1 + 2, (int*)result);
102 CPPUNIT_ASSERT( result[0] == 10 );
103 CPPUNIT_ASSERT( result[1] == 17 );
104 CPPUNIT_ASSERT( result[2] == 23 );
105 CPPUNIT_ASSERT( result[3] == 0 );
106 }
107
setdiff1()108 void SetDifferenceTest::setdiff1()
109 {
110 vector<int> v1(10);
111 __iota(v1.begin(), v1.end(), 0);
112 vector<int> v2(10);
113 __iota(v2.begin(), v2.end(), 7);
114
115 vector<int> diff;
116 set_difference(v1.begin(), v1.end(), v2.begin(), v2.end(), back_inserter(diff));
117 CPPUNIT_ASSERT( diff.size() == 7 );
118 for (int i = 0; i < 7; ++i) {
119 CPPUNIT_ASSERT( diff[i] == i );
120 }
121 }
122
setdiff2()123 void SetDifferenceTest::setdiff2()
124 {
125 const char* word1 = "ABCDEFGHIJKLMNO";
126 const char* word2 = "LMNOPQRSTUVWXYZ";
127
128 string diff;
129 set_difference(word1, word1 + ::strlen(word1), word2, word2 + ::strlen(word2), back_inserter(diff), less<char>());
130 CPPUNIT_ASSERT( diff.size() == 11 );
131 for (int i = 0; i < 11; ++i) {
132 CPPUNIT_ASSERT( diff[i] == ('A' + i) );
133 }
134 }
135