Merge pull request #44 from atcoder/patch/expander

Add tests for expander

Author: yosupo06

.github/workflows/expander.yml

@@ -0,0 +1,23 @@
+name: Expander test
+
+on: [push, pull_request]
+
+jobs:
+  expander:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-latest, windows-latest]
+        python-version: [3.6]
+    steps:
+      - uses: actions/checkout@v2
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Set up pytest
+        run: |
+          python -m pip install --upgrade pip
+          pip install pytest
+      - name: Run test_expander.py
+        run: pytest

.gitignore

@@ -2,4 +2,5 @@ build
 
 *.html
 !tools/*.html
-tools/ac-library.zip
+tools/ac-library.zip
+__pycache__

test/expander/comment_out.cpp

@@ -0,0 +1,13 @@
+// #include <atcoder/dsu>
+/* #include <atcoder/dsu> */
+/* #include <atcoder/dsu>
+*/
+
+namespace atcoder {
+    struct dsu {
+
+    };
+}
+
+int main() {
+}

test/expander/include_all.cpp

@@ -0,0 +1,27 @@
+#include <atcoder/all>
+#include <cstdio>
+
+using namespace std;
+using namespace atcoder;
+
+int main() {
+    int n, q;
+    scanf("%d %d", &n, &q);
+
+    dsu uf(n);
+    for (int i = 0; i < q; i++) {
+        int t, u, v;
+        scanf("%d %d %d", &t, &u, &v);
+        if (t == 0) {
+            // merge
+            uf.merge(u, v);
+        } else {
+            // find
+            if (uf.same(u, v)) {
+                printf("1\n");
+            } else {
+                printf("0\n");
+            }
+        }
+    }
+}

test/expander/include_unionfind.cpp

@@ -0,0 +1,27 @@
+#include <atcoder/dsu>
+#include <cstdio>
+
+using namespace std;
+using namespace atcoder;
+
+int main() {
+    int n, q;
+    scanf("%d %d", &n, &q);
+
+    dsu uf(n);
+    for (int i = 0; i < q; i++) {
+        int t, u, v;
+        scanf("%d %d %d", &t, &u, &v);
+        if (t == 0) {
+            // merge
+            uf.merge(u, v);
+        } else {
+            // find
+            if (uf.same(u, v)) {
+                printf("1\n");
+            } else {
+                printf("0\n");
+            }
+        }
+    }
+}

test/expander/include_unusual_format.cpp

@@ -0,0 +1,28 @@
+#include  <atcoder/dsu>
+#include    <atcoder/dsu>  
+#include <cstdio>
+
+using namespace std;
+using namespace atcoder;
+
+int main() {
+    int n, q;
+    scanf("%d %d", &n, &q);
+
+    dsu uf(n);
+    for (int i = 0; i < q; i++) {
+        int t, u, v;
+        scanf("%d %d %d", &t, &u, &v);
+        if (t == 0) {
+            // merge
+            uf.merge(u, v);
+        } else {
+            // find
+            if (uf.same(u, v)) {
+                printf("1\n");
+            } else {
+                printf("0\n");
+            }
+        }
+    }
+}

test/test_expander.py

@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+
+import unittest
+from logging import Logger, basicConfig, getLogger
+from os import getenv, environ
+from pathlib import Path
+from shutil import copy
+from subprocess import run
+from tempfile import TemporaryDirectory
+from typing import List
+
+logger = getLogger(__name__)  # type: Logger
+
+
+class Test(unittest.TestCase):
+    def compile_test(self, source: Path, env=None):
+        if not env:
+            env = environ.copy()
+        lib_dir = Path.cwd().resolve()
+        expander_path = Path('expander.py').resolve()
+        with TemporaryDirectory() as new_dir:
+            tmp = Path(new_dir)
+            proc = run(['python', str(expander_path), str(
+                source.resolve()), '--lib', str(lib_dir)], cwd=str(tmp), env=env)
+            self.assertEqual(proc.returncode, 0)
+            proc = run(['g++', 'combined.cpp', '-std=c++14'], cwd=str(tmp))
+            self.assertEqual(proc.returncode, 0)
+
+    def test_unionfind(self):
+        self.compile_test(Path('test/expander/include_unionfind.cpp'))
+
+    def test_unusual_format(self):
+        self.compile_test(Path('test/expander/include_unusual_format.cpp'))
+
+    def test_all(self):
+        self.compile_test(Path('test/expander/include_all.cpp'))
+
+    def test_comment_out(self):
+        self.compile_test(Path('test/expander/comment_out.cpp'))
+
+    def test_env_value(self):
+        env = environ.copy()
+        env['CPLUS_INCLUDE_PATH'] = str(Path.cwd().resolve())
+        self.compile_test(Path('test/expander/include_all.cpp'), env=env)
+
+
+if __name__ == "__main__":
+    basicConfig(
+        level=getenv('LOG_LEVEL', 'DEBUG'),
+        format="%(asctime)s %(levelname)s %(name)s : %(message)s"
+    )
+    unittest.main()