* start the coverage earlier

Author: HYLcool

tests/run.py

@@ -12,6 +12,10 @@
 import unittest
 import coverage
 
+# start the coverage immediately
+cov = coverage.Coverage(include='data_juicer/**')
+cov.start()
+
 from loguru import logger
 
 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'):
 
 
 def main():
-    cov = coverage.Coverage(include='data_juicer/**')
-    cov.start()
-
+    global cov
     runner = unittest.TextTestRunner()
     test_suite = gather_test_cases(os.path.abspath(args.test_dir),
                                    args.pattern, args.tag, args.mode)