def test_create_scheduler(self, mock_scheduler): name = 'foo' host = 'localhost' mock_sch = mock.MagicMock() mock_scheduler.return_value = mock_sch self.assertEquals(mock_sch, create_scheduler(name, host)) mock_scheduler.assert_called_once_with('foo', 'localhost', None, None, None, None)
def test_create_scheduler_condor(self, mock_ccs): ret = create_scheduler('foo', 'localhost', 'condor') self.assertEqual(mock_ccs(), ret)
def test_create_scheduler_dask(self, mock_cds): ret = create_scheduler('foo', 'localhost', 'dask') self.assertEqual(mock_cds(), ret)