示例#1
0
    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_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)
 def test_create_scheduler_dask(self, mock_cds):
     ret = create_scheduler('foo', 'localhost', 'dask')
     self.assertEqual(mock_cds(), ret)