def test_emr_runner(self): # EMR runner is always in cluster mode runner = EMRJobRunner() self.assertEqual(runner._logs_needed_to_pick_error('streaming'), ('step', 'history', 'task')) self.assertEqual(runner._logs_needed_to_pick_error('spark'), ('step', 'task'))