Enable tests in CI (#14)

Author: jovany-wang

.github/workflows/ubuntu_basic.yml

@@ -18,7 +18,7 @@ jobs:
     - name: Install bazel
       run:  |
             apt-get update
-            apt-get install -yq wget gcc g++ python3.8 zlib1g-dev zip libuv1.dev
+            apt-get install -yq wget gcc g++ python3.7 zlib1g-dev zip libuv1.dev
             apt-get install -yq pip
             wget "https://github.com/bazelbuild/bazel/releases/download/5.1.0/bazel_5.1.0-linux-x86_64.deb" -O bazel_5.1.0-linux-x86_64.deb
             dpkg -i bazel_5.1.0-linux-x86_64.deb
@@ -32,12 +32,8 @@ jobs:
             pip install pytest torch
             pip install ray==1.11.0
 
-    - name: Build Pygloo
+    - name: Build and test
       run: |
            . py3/bin/activate
-           python setup.py install
-
-    - name: Test
-      run: |
-           . py3/bin/activate
-           pytest -v ./tests
+           python3 setup.py install
+           cd tests && python3 -m pytest *

tests/test_custom_store.py

@@ -5,6 +5,7 @@
 import time
 import shutil
 import torch
+import pytest
 
 import ray.experimental.internal_kv as internal_kv
 from ray._private.gcs_utils import GcsClient
@@ -55,7 +56,7 @@ def __init__(self):
         custom_store = pygloo.rendezvous.CustomStore(real_store)
         self._context.connectFullMesh(custom_store, dev)
 
-    def test_send(self):
+    def do_send(self):
         sendbuf = np.array([[1,2,3],[1,2,3]], dtype=np.float32)
         sendptr = sendbuf.ctypes.data
         pygloo.send(self._context, sendptr, sendbuf.size, pygloo.glooDataType_t.glooFloat32, 1)
@@ -74,7 +75,7 @@ def __init__(self):
         custom_store = pygloo.rendezvous.CustomStore(real_store)
         self._context.connectFullMesh(custom_store, dev)
 
-    def test_recv(self):
+    def do_recv(self):
         recvbuf = np.zeros((2, 3), dtype=np.float32)
         recvptr = recvbuf.ctypes.data
 
@@ -86,19 +87,21 @@ def test_recv(self):
         return recvbuf
 
 
-def main():
+def test_basic():
     ray.init(num_cpus=6)
 
     sender = Sender.remote()
     recver = Recver.remote()
-    fn1 = sender.test_send.remote()
-    fn2 = recver.test_recv.remote()
+    fn1 = sender.do_send.remote()
+    fn2 = recver.do_recv.remote()
 
     a, b = ray.get([fn1, fn2])
-    print(a)
-    print(b)
-    print("END")
+    assert a
+    expected = [[1, 2, 3], [1, 2, 3]]
+    assert len(b) == 2
+    assert len(b[0]) == 3
+    assert len(b[1]) == 3
 
 
 if __name__ == "__main__":
-    main()
+    sys.exit(pytest.main(["-v", __file__]))