[enh] Added tests (py.test)