1 #include <vector> 2 #include <algorithm> 3 #include <functional> 4 5 #include "cppunit/cppunit_proxy.h" 6 7 #if !defined (STLPORT) || defined(_STLP_USE_NAMESPACES) 8 using namespace std; 9 #endif 10 11 // 12 // TestCase class 13 // 14 class NthElemTest : public CPPUNIT_NS::TestCase 15 { 16 CPPUNIT_TEST_SUITE(NthElemTest); 17 CPPUNIT_TEST(nthelem0); 18 CPPUNIT_TEST(nthelem1); 19 CPPUNIT_TEST(nthelem2); 20 CPPUNIT_TEST_SUITE_END(); 21 22 protected: 23 void nthelem0(); 24 void nthelem1(); 25 void nthelem2(); 26 }; 27 28 CPPUNIT_TEST_SUITE_REGISTRATION(NthElemTest); 29 30 // 31 // tests implementation 32 // 33 void NthElemTest::nthelem0() 34 { 35 int numbers[7] = { 5, 2, 4, 1, 0, 3 ,77}; 36 nth_element(numbers, numbers + 3, numbers + 6); 37 38 CPPUNIT_ASSERT(numbers[0]==1); 39 CPPUNIT_ASSERT(numbers[1]==0); 40 CPPUNIT_ASSERT(numbers[2]==2); 41 CPPUNIT_ASSERT(numbers[3]==3); 42 CPPUNIT_ASSERT(numbers[4]==4); 43 CPPUNIT_ASSERT(numbers[5]==5); 44 } 45 void NthElemTest::nthelem1() 46 { 47 //6 8 5 1 7 4 1 5 2 6 48 //1 1 4 2 5 5 6 7 8 6 49 int numbers[10] = { 6, 8, 5, 1, 7, 4, 1, 5, 2, 6 }; 50 51 vector <int> v1(numbers, numbers+10); 52 nth_element(v1.begin(), v1.begin() + v1.size() / 2, v1.end()); 53 54 CPPUNIT_ASSERT(v1[0]==1); 55 CPPUNIT_ASSERT(v1[1]==1); 56 CPPUNIT_ASSERT(v1[2]==4); 57 CPPUNIT_ASSERT(v1[3]==2); 58 CPPUNIT_ASSERT(v1[4]==5); 59 CPPUNIT_ASSERT(v1[5]==5); 60 CPPUNIT_ASSERT(v1[6]==6); 61 CPPUNIT_ASSERT(v1[7]==7); 62 CPPUNIT_ASSERT(v1[8]==8); 63 CPPUNIT_ASSERT(v1[9]==6); 64 } 65 void NthElemTest::nthelem2() 66 { 67 //4 5 4 2 1 7 4 3 1 6 68 //6 7 4 4 5 4 3 2 1 1 69 70 int numbers[10] = { 4, 5, 4, 2, 1, 7, 4, 3, 1, 6 }; 71 vector <int> v1(numbers, numbers+10); 72 nth_element(v1.begin(), v1.begin() + v1.size() / 2, v1.end(), greater<int>()); 73 74 CPPUNIT_ASSERT(v1[0]==6); 75 CPPUNIT_ASSERT(v1[1]==7); 76 CPPUNIT_ASSERT(v1[2]==4); 77 CPPUNIT_ASSERT(v1[3]==4); 78 CPPUNIT_ASSERT(v1[4]==5); 79 CPPUNIT_ASSERT(v1[5]==4); 80 CPPUNIT_ASSERT(v1[6]==3); 81 CPPUNIT_ASSERT(v1[7]==2); 82 CPPUNIT_ASSERT(v1[8]==1); 83 CPPUNIT_ASSERT(v1[9]==1); 84 } 85
