def test_fork(self): sess = mock.MagicMock() body = JSONObject({ 'name': 'ds name', 'description': 'ds description' }) ds = Dataset(session=sess, body=body) ds.forks = mock.MagicMock() ds.forks.index = {} f = ds.fork() ds.forks.create.assert_called_with( { 'body': { 'name': 'FORK #1 of ds name', 'description': 'ds description', 'is_published': False, } } ) f.create_savepoint.assert_called_with('initial fork')