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
