1 #include <vector>
2 #include <numeric>
3 #include <algorithm>
4
5 #include "iota.h"
6 #include "cppunit/cppunit_proxy.h"
7
8 #if !defined (STLPORT) || defined(_STLP_USE_NAMESPACES)
9 using namespace std;
10 #endif
11
12 //
13 // TestCase class
14 //
15 class SearchTest : public CPPUNIT_NS::TestCase
16 {
17 CPPUNIT_TEST_SUITE(SearchTest);
18 CPPUNIT_TEST(search0);
19 CPPUNIT_TEST(search1);
20 CPPUNIT_TEST(search2);
21 CPPUNIT_TEST_SUITE_END();
22
23 protected:
24 void search0();
25 void search1();
26 void search2();
27
str_equal(const char * a_,const char * b_)28 static bool str_equal(const char* a_, const char* b_)
29 {
30 return strcmp(a_, b_) == 0 ? 1 : 0;
31 }
32 };
33
34 CPPUNIT_TEST_SUITE_REGISTRATION(SearchTest);
35
36 //
37 // tests implementation
38 //
search0()39 void SearchTest::search0()
40 {
41 int v1[6] = { 1, 1, 2, 3, 5, 8 };
42 int v2[6] = { 0, 1, 2, 3, 4, 5 };
43 int v3[2] = { 3, 4 };
44
45 int* location;
46 location = search((int*)v1, (int*)v1 + 6, (int*)v3, (int*)v3 + 2);
47 CPPUNIT_ASSERT(location == v1 + 6);
48
49 location = search((int*)v2, (int*)v2 + 6, (int*)v3, (int*)v3 + 2);
50 CPPUNIT_ASSERT(location != v2 + 6);
51 CPPUNIT_ASSERT(location - v2 == 3);
52 }
search1()53 void SearchTest::search1()
54 {
55 typedef vector <int> IntVec;
56 IntVec v1(10);
57 __iota(v1.begin(), v1.end(), 0);
58 IntVec v2(3);
59 __iota(v2.begin(), v2.end(), 50);
60
61 IntVec::iterator location;
62 location = search(v1.begin(), v1.end(), v2.begin(), v2.end());
63
64 CPPUNIT_ASSERT(location == v1.end());
65
66 __iota(v2.begin(), v2.end(), 4);
67
68 location = search(v1.begin(), v1.end(), v2.begin(), v2.end());
69
70 CPPUNIT_ASSERT(location != v1.end());
71 CPPUNIT_ASSERT(location - v1.begin() == 4);
72 }
search2()73 void SearchTest::search2()
74 {
75 char const* grades[] = { "A", "B", "C", "D", "F" };
76 char const* letters[] = { "Q", "E", "D" };
77 const unsigned gradeCount = sizeof(grades) / sizeof(grades[0]);
78 const unsigned letterCount = sizeof(letters) / sizeof(letters[0]);
79 char const** location = search((char const**)grades, (char const**)grades + gradeCount, (char const**)letters, (char const**)letters + letterCount, str_equal);
80
81 CPPUNIT_ASSERT(location == grades + gradeCount);
82
83 copy((char const**)grades + 1, (char const**)grades + 1 + letterCount, (char const**)letters);
84 location = search((char const**)grades, (char const**)grades + gradeCount, (char const**)letters, (char const**)letters + letterCount, str_equal);
85
86 CPPUNIT_ASSERT(location != grades + gradeCount);
87 CPPUNIT_ASSERT(location - grades == 1);
88
89 }
90