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()
|
