comparison tests/test_cmdline_util.py @ 292:9907db0843cb

Refactor main script

 
 _ArgsNamespece = namedtuple(
     "_ArgsNamespece",
     [
         "input",
-        "max_time",
+        "max_read",
         "analysis_window",
         "sampling_rate",
         "sample_width",
         "channels",
         "use_channel",

         "output_format",
         "large_file",
         "frame_per_buffer",
         "input_device_index",
         "save_stream",
+        "save_detections_as",
         "plot",
         "save_image",
         "min_duration",
         "max_duration",
         "max_silence",
         "drop_trailing_silence",
         "strict_min_duration",
         "energy_threshold",
         "echo",
+        "progress_bar",
         "command",
         "quiet",
         "printf",
         "time_format",
         "timestamp_format",
     ],
 )
 
 
 @genty
-class _TestCmdLineUtil(TestCase):
+class TestCmdLineUtil(TestCase):
     @genty_dataset(
-        no_record=("stream.ogg", False, None, "mix", "mix", False),
-        no_record_plot=("stream.ogg", True, None, None, None, False),
+        no_record=("stream.ogg", None, False, None, "mix", "mix", False),
+        no_record_plot=("stream.ogg", None, True, None, None, None, False),
         no_record_save_image=(
             "stream.ogg",
+            None,
             True,
             "image.png",
             None,
             None,
             False,
         ),
-        record_plot=(None, True, None, None, None, True),
-        record_save_image=(None, False, "image.png", None, None, True),
-        int_use_channel=("stream.ogg", False, None, "1", 1, False),
+        record_plot=(None, None, True, None, None, None, True),
+        record_save_image=(None, None, False, "image.png", None, None, True),
+        int_use_channel=("stream.ogg", None, False, None, "1", 1, False),
+        save_detections_as=("stream.ogg", "{id}.wav", False, None, None, None, False)
     )
     def test_make_kwargs(
         self,
         save_stream,
+        save_detections_as,
         plot,
         save_image,
         use_channel,
         exp_use_channel,
         exp_record,

             "ogg",
             True,
             None,
             1,
             save_stream,
+            save_detections_as,
             plot,
             save_image,
             0.2,
             10,
             0.3,
             False,
             False,
             55,
         )
-        misc = (False, None, True, None, "TIME_FORMAT", "TIMESTAMP_FORMAT")
-
+        misc = (False, False, None, True, None, "TIME_FORMAT", "TIMESTAMP_FORMAT")
         args_ns = _ArgsNamespece(*(args + misc))
 
         io_kwargs = {
             "input": "file",
             "max_read": 30,

             "sampling_rate": 16000,
             "sample_width": 2,
             "channels": 2,
             "use_channel": exp_use_channel,
             "save_stream": save_stream,
+            "save_detections_as": save_detections_as,
             "audio_format": "raw",
             "export_format": "ogg",
             "large_file": True,
             "frames_per_buffer": None,
             "input_device_index": 1,

         }
 
         miscellaneous = {
             "echo": False,
             "command": None,
+            "progress_bar": False,
             "quiet": True,
             "printf": None,
             "time_format": "TIME_FORMAT",
             "timestamp_format": "TIMESTAMP_FORMAT",
         }