def test_delete_data():
file_id = "test_id"
fm.delete_data(file_id)
path = fm.get_data_path(file_id)
res = fm.get_data_info(path)
eq_(res, {})
def test_get_data_path():
file_id = "test_id"
path = fm.get_data_path(file_id)
ok_(isinstance(path, Path))
eq_(path.parent, Path(fm.data_dir))
eq_(path.name, file_id)
def test_get_data_info_4():
file_id = "no_info"
path = fm.get_data_path(file_id)
res = fm.get_data_info(path)
expected = {
'id': 'no_info',
'nImages': 4,
'nLabels': 4,
'nClasses': 0,
'statistics': {},
'name': '',
'description': '',
'update_time': '2018-08-11 23:32:41',
'create_time': '2018-08-11 23:32:58'
}
eq_(res, expected)
def test_get_data_info_2():
file_id = "test001"
path = fm.get_data_path(file_id)
res = fm.get_data_info(path)
expected = {
'id': 'test001',
'nImages': 4,
'nLabels': 4,
'nClasses': 3,
'statistics': {'nameA': 2, 'nameB': 1, 'nameC': 1},
'name': 'fashion MNIST',
'description': 'fashion MNIST',
'update_time': '2018-08-10 16:52:25',
'create_time': '2018-08-10 15:45:32'
}
eq_(res, expected)
def test_get_data_info_1():
file_id = "test_id"
path = fm.get_data_path(file_id)
res = fm.get_data_info(path)
new_data = {"name": "aaaa", "description": "aaaa"}
info_path = path / "info" / "info.json"
expected = {
'id': file_id,
'nTexts': 0,
'nImages': 0,
'nLabels': 0,
'nClasses': 0,
'statistics': {},
'name': new_data["name"],
'description': new_data["description"],
'update_time': fm.get_update_time(info_path),
'create_time': fm.get_create_time(path)
}
eq_(res, expected)
def test_update_data_info():
file_id = "test_id"
update_data = {"name": "bbbb", "description": "bbbb"}
res = fm.update_data_info(update_data, file_id)
data_path = fm.get_data_path(file_id)
info_path = data_path / "info" / "info.json"
expected = {
'id': file_id,
'nTexts': 0,
'nImages': 0,
'nLabels': 0,
'nClasses': 0,
'statistics': {},
'name': update_data["name"],
'description': update_data["description"],
'update_time': fm.get_update_time(info_path),
'create_time': fm.get_create_time(data_path)
}
eq_(res, expected)
def test_put_data_info():
new_data = {"name": "aaaa", "description": "aaaa"}
file_id = "test_id"
res = fm.put_data_info(new_data, file_id)
data_path = fm.get_data_path(file_id)
info_path = data_path / "info" / "info.json"
expected = {
'id': file_id,
'nTexts': 0,
'nImages': 0,
'nLabels': 0,
'nClasses': 0,
'statistics': {},
'name': new_data["name"],
'description': new_data["description"],
'update_time': fm.get_update_time(info_path),
'create_time': fm.get_create_time(data_path)
}
eq_(expected, res)
def test_get_data_info_3():
file_id = "not_exist_id"
path = fm.get_data_path(file_id)
res = fm.get_data_info(path)
eq_(res, {})
def test_get_images():
file_id = "test001"
path = fm.get_data_path(file_id)
images_path = path / "images"
res = fm.get_images(images_path)
eq_(len(res), 4)
