Python _choose_backendの例

コード例 #1

ファイル: test_sampling.py プロジェクト: DukasGuo/pymc3

 def test_choose_backend_shortcut(self):
     backend = mock.Mock()
     shortcuts = {'test_backend': {'backend': backend,
                                   'name': None}}
     sampling._choose_backend('test_backend', 'chain', shortcuts=shortcuts)
     self.assertTrue(backend.called)

コード例 #2

ファイル: test_sampling.py プロジェクト: DukasGuo/pymc3

 def test_choose_backend_list_of_variables(self):
     with mock.patch('pymc3.sampling.NDArray') as nd:
         sampling._choose_backend(['var1', 'var2'], 'chain')
     nd.assert_called_with(vars=['var1', 'var2'])

コード例 #3

ファイル: test_sampling.py プロジェクト: DukasGuo/pymc3

 def test_choose_backend_none(self):
     with mock.patch('pymc3.sampling.NDArray') as nd:
         sampling._choose_backend(None, 'chain')
     self.assertTrue(nd.called)

コード例 #4

ファイル: test_sampling.py プロジェクト: zjlearn/pymc3

 def test_choose_backend_shortcut(self):
     backend = mock.Mock()
     shortcuts = {'test_backend': {'backend': backend, 'name': None}}
     sampling._choose_backend('test_backend', 'chain', shortcuts=shortcuts)
     self.assertTrue(backend.called)

コード例 #5

ファイル: test_sampling.py プロジェクト: zjlearn/pymc3

 def test_choose_backend_list_of_variables(self):
     with mock.patch('pymc3.sampling.NDArray') as nd:
         sampling._choose_backend(['var1', 'var2'], 'chain')
     nd.assert_called_with(vars=['var1', 'var2'])

コード例 #6

ファイル: test_sampling.py プロジェクト: zjlearn/pymc3

 def test_choose_backend_none(self):
     with mock.patch('pymc3.sampling.NDArray') as nd:
         sampling._choose_backend(None, 'chain')
     self.assertTrue(nd.called)