mirror of
https://github.com/microsoft/qlib.git
synced 2026-06-06 05:51:17 +08:00
28 lines
684 B
Python
28 lines
684 B
Python
# Copyright (c) Microsoft Corporation.
|
|
# Licensed under the MIT License.
|
|
|
|
import unittest
|
|
|
|
from qlib.contrib.model import all_model_classes
|
|
|
|
|
|
class TestAllFlow(unittest.TestCase):
|
|
def test_0_initialize(self):
|
|
num = 0
|
|
for model_class in all_model_classes:
|
|
if model_class is not None:
|
|
model = model_class()
|
|
num += 1
|
|
print("There are {:}/{:} valid models in total.".format(num, len(all_model_classes)))
|
|
|
|
|
|
def suite():
|
|
_suite = unittest.TestSuite()
|
|
_suite.addTest(TestAllFlow("test_0_initialize"))
|
|
return _suite
|
|
|
|
|
|
if __name__ == "__main__":
|
|
runner = unittest.TextTestRunner()
|
|
runner.run(suite())
|