model/third_party/HMNet/DataLoader/infinibatch/test/test_doctests.py

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

"""
This file causes the doctests to be included as part of unit tests.

To make sure the doctests of a specific module are included,
please replicate the `addTests` call for the iterators module below.
"""

import doctest
import infinibatch.iterators


def load_tests(loader, tests, ignore):
    tests.addTests(doctest.DocTestSuite(infinibatch.iterators))
    return tests