From 20357998176a8a65b0e910eba4e55921f7fe872c Mon Sep 17 00:00:00 2001
From: tqchen <tianqi.tchen@gmail.com>
Date: Fri, 20 Mar 2015 13:02:46 -0700
Subject: [PATCH] test code

---
 rabit-learn/io/hdfs-inl.h       | 14 +++++++++++++-
 rabit-learn/linear/test_load.cc |  8 ++++----
 2 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/rabit-learn/io/hdfs-inl.h b/rabit-learn/io/hdfs-inl.h
index 967f7b5c0..cf5c3f599 100644
--- a/rabit-learn/io/hdfs-inl.h
+++ b/rabit-learn/io/hdfs-inl.h
@@ -23,6 +23,7 @@ class HDFSStream : public ISeekStream {
              bool disconnect_when_done)
       : fs_(fs), at_end_(false),
         disconnect_when_done_(disconnect_when_done) {
+    fsbk_ = fs_;
     int flag = 0;
     if (!strcmp(mode, "r")) {
       flag = O_RDONLY;
@@ -44,6 +45,7 @@ class HDFSStream : public ISeekStream {
     }
   }
   virtual size_t Read(void *ptr, size_t size) {
+    CheckFS();
     tSize nread = hdfsRead(fs_, fp_, ptr, size);
     if (nread == -1) {
       int errsv = errno;
@@ -55,6 +57,7 @@ class HDFSStream : public ISeekStream {
     return static_cast<size_t>(nread);
   }
   virtual void Write(const void *ptr, size_t size) {
+    CheckFS();
     const char *buf = reinterpret_cast<const char*>(ptr);
     while (size != 0) {
       tSize nwrite = hdfsWrite(fs_, fp_, buf, size);
@@ -67,12 +70,14 @@ class HDFSStream : public ISeekStream {
     }
   }
   virtual void Seek(size_t pos) {
+    CheckFS();
     if (hdfsSeek(fs_, fp_, pos) != 0) {
       int errsv = errno;
       utils::Error("HDFSStream.Seek Error:%s", strerror(errsv));
     }
   }
   virtual size_t Tell(void) {
+    CheckFS();
     tOffset offset = hdfsTell(fs_, fp_);
     if (offset == -1) {
       int errsv = errno;
@@ -84,6 +89,7 @@ class HDFSStream : public ISeekStream {
     return at_end_;
   }
   inline void Close(void) {
+    CheckFS();
     if (fp_ != NULL) {
       if (hdfsCloseFile(fs_, fp_) == -1) {
         int errsv = errno;
@@ -93,11 +99,17 @@ class HDFSStream : public ISeekStream {
     }
   }  
   
- private:  
+ private:
+  inline void CheckFS(void) const {
+    if (fs_ != fsbk_) {
+      rabit::TrackerPrintf("[%d] fs flag inconstent\n", rabit::GetRank());
+    }
+  }
   hdfsFS fs_;
   hdfsFile fp_;
   bool at_end_;
   bool disconnect_when_done_;
+  hdfsFS fsbk_;
 };
 
 /*! \brief line split from normal file system */
diff --git a/rabit-learn/linear/test_load.cc b/rabit-learn/linear/test_load.cc
index 05597e503..dc486d6e5 100644
--- a/rabit-learn/linear/test_load.cc
+++ b/rabit-learn/linear/test_load.cc
@@ -3,11 +3,11 @@
 
 int main(int argc, char *argv[]) {
   using namespace rabit::io;
-  if (argc < 2) {
+  if (argc < 4) {
     // intialize rabit engine
     rabit::Init(argc, argv);
     if (rabit::GetRank() == 0) {
-      rabit::TrackerPrintf("Usage: <data_in> param=val\n");
+      rabit::TrackerPrintf("Usage: <data_in> npart rank\n");
     }
     rabit::Finalize();
     return 0;
@@ -15,8 +15,8 @@ int main(int argc, char *argv[]) {
   rabit::Init(argc, argv);
   int n = 0;
   InputSplit *in = CreateInputSplit(argv[1],
-                                    rabit::GetRank(),
-                                    rabit::GetWorldSize());
+                                    atoi(argv[2]),
+                                    atoi(argv[3]));
   std::string line;
   while (in->NextLine(&line)) {
     if (n % 100 == 0) {