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)