TestCase/unittest/catalan.unittest.test.py

View this file on GitHub
Last update: 2021-09-04 15:01:27+09:00
Problem: http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=ITP1_1_A

Depends on

カタラン数 (Combination/catalan.py)

Code

# verification-helper: PROBLEM http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=ITP1_1_A
import sys
input = sys.stdin.buffer.readline

import itertools
from Combination.catalan import catalan, catalan_trapezoid


def naive_catalan(n):
    """DFS naive calculation"""
    def dfs(cnt_pl, cnt_mi):
        if cnt_pl > n or cnt_mi > n or cnt_pl < cnt_mi:
            return 0
        if cnt_pl == n and cnt_mi == n:
            return 1
        return dfs(cnt_pl + 1, cnt_mi) + dfs(cnt_pl, cnt_mi + 1)

    return dfs(0, 0)


def naive_catalan_trapezoid(n, k, m):
    """DFS naive calculation"""
    def dfs(cnt_pl, cnt_mi):
        if cnt_pl > n or cnt_mi > k or cnt_pl + (m - 1) < cnt_mi:
            return 0
        if cnt_pl == n and cnt_mi == k:
            return 1
        return dfs(cnt_pl + 1, cnt_mi) + dfs(cnt_pl, cnt_mi + 1)

    return dfs(0, 0)


def main():
    for n in range(10):
        assert(catalan(n) == naive_catalan(n))

    for n in range(5):
        for k in range(5):
            for m in range(5):
                assert(catalan_trapezoid(n, k, m)
                       == naive_catalan_trapezoid(n, k, m))


if __name__ == '__main__':
    main()
    print("Hello World")

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.12.4/x64/lib/python3.12/site-packages/onlinejudge_verify/documentation/build.py", line 71, in _render_source_code_stat
    bundled_code = language.bundle(stat.path, basedir=basedir, options={'include_paths': [basedir]}).decode()
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.4/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/python.py", line 96, in bundle
    raise NotImplementedError
NotImplementedError