File size: 448 Bytes
546a9ba
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# 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