tests/test_docs.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

import doctest
import inspect
import unittest

import sqlglot
import sqlglot.optimizer
import sqlglot.transforms


def load_tests(loader, tests, ignore):
    """
    This finds and runs all the doctests
    """

    modules = {
        mod
        for module in [sqlglot, sqlglot.transforms, sqlglot.optimizer]
        for _, mod in inspect.getmembers(module, inspect.ismodule)
    }

    assert len(modules) >= 20

    for module in modules:
        tests.addTests(doctest.DocTestSuite(module))

    return tests


if __name__ == "__main__":
    unittest.main()