Skip to content

Commit a774769

Browse files
committed
* start the coverage earlier
1 parent aa51ec0 commit a774769

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/run.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,10 @@
1212
import unittest
1313
import coverage
1414

15+
# start the coverage immediately
16+
cov = coverage.Coverage(include='data_juicer/**')
17+
cov.start()
18+
1519
from loguru import logger
1620

1721
from data_juicer.utils.unittest_utils import set_clear_model_flag, get_partial_test_cases
@@ -91,9 +95,7 @@ def gather_test_cases(test_dir, pattern, tag, mode='partial'):
9195

9296

9397
def main():
94-
cov = coverage.Coverage(include='data_juicer/**')
95-
cov.start()
96-
98+
global cov
9799
runner = unittest.TextTestRunner()
98100
test_suite = gather_test_cases(os.path.abspath(args.test_dir),
99101
args.pattern, args.tag, args.mode)

0 commit comments

Comments
 (0)