add: test suite

Author: Essoz

traincheck/collect_trace.py

@@ -425,6 +425,11 @@ def main():
                 f"Setting warm up steps to default value {config.DEFAULT_CHECKING_POLICY['warm_up']}"
             )
             args.warm_up_steps = config.DEFAULT_CHECKING_POLICY["warm_up"]
+    else:
+        if args.sampling_interval is None:
+            args.sampling_interval = config.INSTRUMENTATION_POLICY["interval"]
+        if args.warm_up_steps is None:
+            args.warm_up_steps = config.INSTRUMENTATION_POLICY["warm_up"]
 
     # set up logging
     if args.debug_mode:

traincheck/instrumentor/tracer.py

@@ -594,14 +594,6 @@ def __init__(
         The user should not need to interact with this file directly.
 
         """
-        if sampling_interval:
-            if config.INSTRUMENTATION_POLICY is None:
-                config.INSTRUMENTATION_POLICY = {}
-            config.INSTRUMENTATION_POLICY["interval"] = sampling_interval
-        if warm_up_steps is not None:
-            if config.INSTRUMENTATION_POLICY is None:
-                config.INSTRUMENTATION_POLICY = {}
-            config.INSTRUMENTATION_POLICY["warm_up"] = warm_up_steps
 
         self.instrumenting = True
         if isinstance(target, types.ModuleType):