feat: add MurmurHash2 hash() implementation test (#355)

Author: Fengzdadi

tests/test_hash.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+# --coding:utf-8--
+
+# Copyright (c) 2020 vesoft inc. All rights reserved.
+#
+# This source code is licensed under Apache 2.0 License.
+
+import pytest
+from nebula3.utils.hash import hash as murmur_hash
+
+TEST_VECTORS = [
+    (b"", 6142509188972423790),
+    (b"a", 4993892634952068459),
+    (b"abcdefgh", 8664279048047335611),  # length-8 cases
+    (b"abcdefghi", -5409788147785758033),
+    ("to_be_hashed", -1098333533029391540),
+    ("中文", -8591787916246384322),
+]
+
+@pytest.mark.parametrize("data, expected", TEST_VECTORS)
+def test_known_vectors(data, expected):
+    assert murmur_hash(data) == expected
+
+
+def test_str_bytes_equiv():
+    """
+    Ensure str and bytes inputs produce the same hash.
+    """
+    s = "pytest"
+    assert murmur_hash(s) == murmur_hash(s.encode("utf-8"))
+
+
+def test_type_error():
+    """
+    TypeError
+    """
+    with pytest.raises(TypeError):
+        murmur_hash(12345)

tests/test_hash_integration.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+# --coding:utf-8--
+
+# Copyright (c) 2020 vesoft inc. All rights reserved.
+#
+# This source code is licensed under Apache 2.0 License.
+
+import pytest
+from nebula3.Config import Config
+from nebula3.gclient.net import ConnectionPool
+from nebula3.utils.hash import hash as murmur_hash
+
+@pytest.fixture(scope="module")
+def nebula_session():
+    config = Config()
+    config.max_connection_pool_size = 10
+    pool = ConnectionPool()
+    pool.init([("127.0.0.1", 9669)], config)
+    session = pool.get_session('root', 'nebula')
+    yield session
+    pool.close()
+
+@pytest.mark.parametrize("data", [
+    "", "a", "abcdefgh", "abcdefghi", "to_be_hashed", "中文"
+])
+def test_hash_against_server(nebula_session, data):
+    # Local Computing
+    expected = murmur_hash(data)
+    result = nebula_session.execute(f'YIELD hash("{data}")')
+    assert result.is_succeeded(), result.error_msg()
+    actual = result.row_values(0)[0].as_int()
+    assert actual == expected