diff --git a/tests/README.md b/tests/README.md
new file mode 100644
index 000000000..19e34d5df
--- /dev/null
+++ b/tests/README.md
@@ -0,0 +1 @@
+This folder contains tetstcases for xgboost.
\ No newline at end of file
diff --git a/tests/python/test_basic.py b/tests/python/test_basic.py
new file mode 100644
index 000000000..77d19595b
--- /dev/null
+++ b/tests/python/test_basic.py
@@ -0,0 +1,31 @@
+import numpy as np
+import xgboost as xgb
+
+dpath = 'demo/data/'
+
+def test_basic():
+    dtrain = xgb.DMatrix(dpath + 'agaricus.txt.train')
+    dtest = xgb.DMatrix(dpath + 'agaricus.txt.test')
+    param = {'max_depth':2, 'eta':1, 'silent':1, 'objective':'binary:logistic' }
+    # specify validations set to watch performance
+    watchlist  = [(dtest,'eval'), (dtrain,'train')]
+    num_round = 2
+    bst = xgb.train(param, dtrain, num_round, watchlist)
+    # this is prediction
+    preds = bst.predict(dtest)
+    labels = dtest.get_label()
+    err = sum(1 for i in range(len(preds)) if int(preds[i]>0.5)!=labels[i]) / float(len(preds))
+    # error must be smaller than 10%
+    assert err < 0.1
+
+    # save dmatrix into binary buffer
+    dtest.save_binary('dtest.buffer')
+    # save model
+    bst.save_model('xgb.model')
+    # load model and data in
+    bst2 = xgb.Booster(model_file='xgb.model')
+    dtest2 = xgb.DMatrix('dtest.buffer')
+    preds2 = bst2.predict(dtest2)
+    # assert they are the same
+    assert np.sum(np.abs(preds2-preds)) == 0
+
diff --git a/wrapper/xgboost.py b/wrapper/xgboost.py
index c21545b4c..a009ad81b 100644
--- a/wrapper/xgboost.py
+++ b/wrapper/xgboost.py
@@ -6,7 +6,7 @@ Version: 0.40
 Authors: Tianqi Chen, Bing Xu
 Early stopping by Zygmunt Zając
 """
-# pylint: disable=too-many-arguments, too-many-locals, too-many-lines
+# pylint: disable=too-many-arguments, too-many-locals, too-many-lines, invalid-name
 from __future__ import absolute_import
 
 import os