Add version() to pyffish

diff --git a/setup.py b/setup.py
index dc57802..96761c3 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -37,7 +37,7 @@ pyffish_module = Extension(
     sources=sources,
     extra_compile_args=args)
 
-setup(name="pyffish", version="0.0.50",
+setup(name="pyffish", version="0.0.51",
       description="Fairy-Stockfish Python wrapper",
       long_description=long_description,
       long_description_content_type="text/markdown",

diff --git a/src/pyffish.cpp b/src/pyffish.cpp
index f2d91d5..feae8ab 100644 (file)
--- a/src/pyffish.cpp
+++ b/src/pyffish.cpp
@@ -49,6 +49,10 @@ void buildPosition(Position& pos, StateListPtr& states, const char *variant, con
     return;
 }
 
+extern "C" PyObject* pyffish_version(PyObject* self) {
+    return Py_BuildValue("(iii)", 0, 0, 51);
+}
+
 extern "C" PyObject* pyffish_info(PyObject* self) {
     return Py_BuildValue("s", engine_info().c_str());
 }
@@ -286,6 +290,7 @@ extern "C" PyObject* pyffish_hasInsufficientMaterial(PyObject* self, PyObject *a
 
 
 static PyMethodDef PyFFishMethods[] = {
+    {"version", (PyCFunction)pyffish_version, METH_NOARGS, "Get package version."},
     {"info", (PyCFunction)pyffish_info, METH_NOARGS, "Get Stockfish version info."},
     {"set_option", (PyCFunction)pyffish_setOption, METH_VARARGS, "Set UCI option."},
     {"start_fen", (PyCFunction)pyffish_startFen, METH_VARARGS, "Get starting position FEN."},

diff --git a/test.py b/test.py
index 0fee247..93f5d7a 100644 (file)
--- a/test.py
+++ b/test.py
@@ -123,6 +123,10 @@ variant_positions = {
 
 
 class TestPyffish(unittest.TestCase):
+    def test_version(self):
+        result = sf.version()
+        self.assertEqual(result, (0, 0, 51))
+
     def test_info(self):
         result = sf.info()
         self.assertTrue(result.startswith("Fairy-Stockfish"))