Changeset 879 for trunk/test

Timestamp:

Sep 21, 2007, 4:10:07 PM (16 years ago)

Author:

Markus Ringnér

Message:

Added support to utility::vector for gsl-functions that finds indices to the k largest/smallest elements in a vector

File:

• trunk/test/vector_test.cc (modified) (2 diffs)

trunk/test/vector_test.cc

@@ -242 +242 @@
   } catch (utility::IO_error& err) {
     *message << err.what() << std::endl;
-    this_ok=true; // good, exceoption thrown, test passed
+    this_ok=true; // good, exception thrown, test passed
   }
   try {
@@ -267 +267 @@
   ok &= this_ok;
 
-  std::string data3("data/vector3.data");
-  std::ifstream data_stream3(data3.c_str());
-  utility::vector vec3(data_stream3);
-  std::vector<size_t> dummy;
-  utility::sort_index(dummy,vec3);
-  for(size_t i=0;i<dummy.size();i++) {
-    *message << "dummy: " << dummy[i] << std::endl;
+
+  // Test sort algorithms
+  {
+    std::string data3("data/vector3.data");
+    std::ifstream data_stream3(data3.c_str());
+    utility::vector vec3(data_stream3);
+    std::vector<size_t> dummy;
+    dummy.push_back(100); // To make sure it works starting with a non-empty vector
+
+    utility::sort_index(dummy,vec3);
+    if(dummy.size()!=vec3.size()) {
+      ok=false;
+      *message << "Vector of sorted indices has incorrect size" << std::endl;
+    }
+    if(dummy[0]!=11 || dummy[dummy.size()-1]!=8) {
+      ok=false;
+      *message << "Vector of sorted indices incorrect" << std::endl;
+    }
+    size_t k=3;
+
+    utility::sort_smallest_index(dummy,k,vec3);
+    if(dummy.size()!=k) {
+      ok=false;
+      *message << "Vector of sorted smallest indices has incorrect size" << std::endl;
+    }
+    if(dummy[0]!=11 || dummy[dummy.size()-1]!=9) {
+      ok=false;
+      *message << "Vector of sorted smallest indices incorrect" << std::endl;
+    }
+
+    k=4;
+    utility::sort_largest_index(dummy,k,vec3);
+    if(dummy.size()!=k) {
+      ok=false;
+      *message << "Vector of sorted largest indices has incorrect size" << std::endl;
+    }
+    if(dummy[0]!=8 || dummy[dummy.size()-1]!=5) {
+      ok=false;
+      *message << "Vector of sorted largest indices incorrect" << std::endl;
+    }
   }