def test_remove_categories():
ds = CocoDataset(BASE_PATH / 'new_coco.json')
cat_id = ds.add_category('dog', 'animal')
assert len(ds.cats) == 1
ds.remove_categories([cat_id])
assert len(ds.cats) == 0
def test_remove_categories():
ds = CocoDataset(BASE_PATH / "new_coco.json")
cat_id = ds.add_category("dog", "animal")
assert len(ds.cats) == 1
ds.remove_categories([cat_id])
assert len(ds.cats) == 0
def test_remove_categories_and_annotations():
ds = CocoDataset(BASE_PATH / 'new_coco.json')
cat_id = ds.add_category('dog', 'animal')
img_id = ds.add_image((BASE_PATH / 'test_nodamage.jpg').as_posix())
ds.add_annotation(img_id, cat_id, [1, 2, 3, 4, 5], 10, [0, 0, 256, 256], 0)
assert len(ds.cats) == 1
ds.remove_categories([cat_id])
assert len(ds.cats) == 0
def test_remove_categories_and_annotations():
ds = CocoDataset(BASE_PATH / "new_coco.json")
cat_id = ds.add_category("dog", "animal")
img_id = ds.add_image((BASE_PATH / "test_nodamage.jpg").as_posix(), 100,
100)
ds.add_annotation(img_id, cat_id, [1, 2, 3, 4, 5], 10, [0, 0, 256, 256], 0)
assert len(ds.cats) == 1
ds.remove_categories([cat_id])
assert len(ds.cats) == 0
