def test_load_empty_savepoint(self): sess = mock.MagicMock() ds = Dataset(session=sess) ds.savepoints = mock.MagicMock() ds.savepoints.index = {} with pytest.raises(KeyError) as err: ds.load_savepoint('savepoint')