def test_forks_dataframe(self): f1 = dict( name='name', description='description', is_published=True, owner_name='Jane Doe', current_editor_name='John Doe', creation_time='2016-01-01T00:00Z', modification_time='2016-01-01T00:00Z', id='abc123', ) sess = mock.MagicMock() ds = Dataset(session=sess) ds.forks = mock.MagicMock() ds.forks.index = { 'abc1': f1 } df = ds.forks_dataframe() assert isinstance(df, DataFrame) keys = [k for k in df.keys()] assert keys == [ 'name', 'description', 'is_published', 'owner_name', 'current_editor_name', 'creation_time', 'modification_time', 'id' ]
def test_forks_dataframe_empty(self): sess = mock.MagicMock() ds = Dataset(session=sess) ds.forks = mock.MagicMock() ds.forks.index = {} df = ds.forks_dataframe() assert df is None