code: def test__work(mock__get_atm):
      # setup
      args_mock = Mock(
          dataruns=[1],
          total_time=[1],
          save_files=False,
          cloud_mode=False,
      )
  
      # run
      cli._work(args_mock)
  
      # assert
      mock__get_atm.assert_called_once_with(args_mock)
  
      mock__get_atm.return_value.work.assert_called_once_with(
          datarun_ids=[1],
          choose_randomly=False,
          save_files=False,
          cloud_mode=False,
          total_time=[1],
          wait=False,
      )
  
file_path: ATM/tests/test_cli.py
imports: from mock import Mock, patch
  from atm import cli
label_desc: Use image node_id 5 for calling a global function with example usage: test__work(mock__get_atm) without return types