sl@0: #include <vector>
sl@0: #include <numeric>
sl@0: #include <algorithm>
sl@0: 
sl@0: #include "iota.h"
sl@0: #include "cppunit/cppunit_proxy.h"
sl@0: 
sl@0: #if !defined (STLPORT) || defined(_STLP_USE_NAMESPACES)
sl@0: using namespace std;
sl@0: #endif
sl@0: 
sl@0: //
sl@0: // TestCase class
sl@0: //
sl@0: class SearchTest : public CPPUNIT_NS::TestCase
sl@0: {
sl@0:   CPPUNIT_TEST_SUITE(SearchTest);
sl@0:   CPPUNIT_TEST(search0);
sl@0:   CPPUNIT_TEST(search1);
sl@0:   CPPUNIT_TEST(search2);
sl@0:   CPPUNIT_TEST_SUITE_END();
sl@0: 
sl@0: protected:
sl@0:   void search0();
sl@0:   void search1();
sl@0:   void search2();
sl@0: 
sl@0:   static bool str_equal(const char* a_, const char* b_)
sl@0:   {
sl@0:     return strcmp(a_, b_) == 0 ? 1 : 0;
sl@0:   }
sl@0: };
sl@0: 
sl@0: CPPUNIT_TEST_SUITE_REGISTRATION(SearchTest);
sl@0: 
sl@0: //
sl@0: // tests implementation
sl@0: //
sl@0: void SearchTest::search0()
sl@0: {
sl@0:   int v1[6] = { 1, 1, 2, 3, 5, 8 };
sl@0:   int v2[6] = { 0, 1, 2, 3, 4, 5 };
sl@0:   int v3[2] = { 3, 4 };
sl@0: 
sl@0:   int* location;
sl@0:   location = search((int*)v1, (int*)v1 + 6, (int*)v3, (int*)v3 + 2);
sl@0:   CPPUNIT_ASSERT(location == v1 + 6);
sl@0: 
sl@0:   location = search((int*)v2, (int*)v2 + 6, (int*)v3, (int*)v3 + 2);
sl@0:   CPPUNIT_ASSERT(location != v2 + 6);
sl@0:   CPPUNIT_ASSERT(location - v2 == 3);
sl@0: }
sl@0: void SearchTest::search1()
sl@0: {
sl@0:   typedef vector <int> IntVec;
sl@0:   IntVec v1(10);
sl@0:   __iota(v1.begin(), v1.end(), 0);
sl@0:   IntVec v2(3);
sl@0:   __iota(v2.begin(), v2.end(), 50);
sl@0: 
sl@0:   IntVec::iterator location;
sl@0:   location = search(v1.begin(), v1.end(), v2.begin(), v2.end());
sl@0: 
sl@0:   CPPUNIT_ASSERT(location == v1.end());
sl@0: 
sl@0:   __iota(v2.begin(), v2.end(), 4);
sl@0: 
sl@0:   location = search(v1.begin(), v1.end(), v2.begin(), v2.end());
sl@0: 
sl@0:   CPPUNIT_ASSERT(location != v1.end());
sl@0:   CPPUNIT_ASSERT(location - v1.begin() == 4);
sl@0: }
sl@0: void SearchTest::search2()
sl@0: {
sl@0:   char const* grades[] = { "A", "B", "C", "D", "F" };
sl@0:   char const* letters[] = { "Q", "E", "D" };
sl@0:   const unsigned gradeCount = sizeof(grades) / sizeof(grades[0]);
sl@0:   const unsigned letterCount = sizeof(letters) / sizeof(letters[0]);
sl@0:   char const** location = search((char const**)grades, (char const**)grades + gradeCount, (char const**)letters, (char const**)letters + letterCount, str_equal);
sl@0: 
sl@0:   CPPUNIT_ASSERT(location == grades + gradeCount);
sl@0: 
sl@0:   copy((char const**)grades + 1, (char const**)grades + 1 + letterCount, (char const**)letters);
sl@0:   location = search((char const**)grades, (char const**)grades + gradeCount, (char const**)letters, (char const**)letters + letterCount, str_equal);
sl@0: 
sl@0:   CPPUNIT_ASSERT(location != grades + gradeCount);
sl@0:   CPPUNIT_ASSERT(location - grades == 1);
sl@0: 
sl@0: }