wehub-resource-sync
26 lines
850 B
Python
26 lines
850 B
Python
from cleanlab.datalab.internal.adapter.imagelab import create_imagelab
|
|
from cleanlab.datalab.internal.adapter.constants import DEFAULT_CLEANVISION_ISSUES
|
|
|
|
|
|
class TestImagelabAdapater:
|
|
def test_create_imagelab(self, image_dataset):
|
|
imagelab = create_imagelab(image_dataset, "image")
|
|
assert imagelab is not None
|
|
assert hasattr(imagelab, "issues")
|
|
assert hasattr(imagelab, "issue_summary")
|
|
assert hasattr(imagelab, "info")
|
|
|
|
def test_imagelab_default_issue_types(self):
|
|
default_issues = DEFAULT_CLEANVISION_ISSUES
|
|
assert set(default_issues) == set(
|
|
[
|
|
"dark",
|
|
"light",
|
|
"low_information",
|
|
"odd_aspect_ratio",
|
|
"odd_size",
|
|
"grayscale",
|
|
"blurry",
|
|
]
|
|
)
|