use pytest

rocksdb/merge_operators.py

@@ -0,0 +1,22 @@
+import struct as py_struct
+from interfaces import AssociativeMergeOperator
+
+class UintAddOperator(AssociativeMergeOperator):
+    def merge(self, key, existing_value, value):
+        if existing_value:
+            s = py_struct.unpack('Q', existing_value)[0] + py_struct.unpack('Q', value)[0]
+            return (True, py_struct.pack('Q', s))
+        return (True, value)
+
+    def name(self):
+        return b'uint64add'
+
+class StringAppendOperator(AssociativeMergeOperator):
+    def merge(self, key, existing_value, value):
+        if existing_value:
+            s = existing_value + ',' + value
+            return (True, s)
+        return (True, value)
+
+    def name(self):
+        return b'StringAppendOperator'

setup.py

@@ -43,6 +43,7 @@ setup(
     package_dir={'rocksdb': 'rocksdb'},
     packages=find_packages('.'),
     ext_modules=cythonize([mod1]),
-    test_suite='rocksdb.tests',
+    setup_requires=['pytest-runner'],
+    tests_require=['pytest'],
     include_package_data=True
 )