diff --git a/responsibleai_vision/tests/test_feature_extractors.py b/responsibleai_vision/tests/test_feature_extractors.py index 62206aadb6..78e57d20b7 100644 --- a/responsibleai_vision/tests/test_feature_extractors.py +++ b/responsibleai_vision/tests/test_feature_extractors.py @@ -14,17 +14,22 @@ MEAN_PIXEL_VALUE = ExtractedFeatures.MEAN_PIXEL_VALUE.value FRIDGE_METADATA_FEATURES = [ 'SensingMethod', 'GPSVersionID', 'ISOSpeedRatings', 'SceneType', - 'SceneCaptureType', 'SubjectDistance', 'CustomRendered', 'SubjectDistanceRange', - 'DigitalZoomRatio', 'ApertureValue', 'ImageWidth', 'GPSDOP', 'MaxApertureValue', - 'ColorSpace', 'FocalLengthIn35mmFilm', 'ExposureMode', 'Saturation', 'ExposureTime', - 'ExifImageHeight', 'FNumber', 'YCbCrPositioning', 'Make', 'MeteringMode', 'ExposureBiasValue', - 'ExposureProgram', 'ComponentsConfiguration', 'ExifImageWidth', 'ExifInteroperabilityOffset', - 'BrightnessValue', 'ImageLength', 'FlashPixVersion', 'SubsecTimeOriginal', 'Model', - 'SubsecTimeDigitized', 'ResolutionUnit', 'DateTimeOriginal', 'XResolution', 'FocalLength', - 'Sharpness', 'GPSLongitude', 'Contrast', 'Software', 'GPSLatitude', 'MakerNote', 'GPSDateStamp', - 'GPSAltitude', 'GPSProcessingMethod', 'GPSTimeStamp', 'GPSLatitudeRef', 'WhiteBalance', - 'GPSLongitudeRef', 'Flash', 'SubsecTime', 'YResolution', 'DateTimeDigitized', 'DateTime', - 'GPSAltitudeRef', 'Orientation', 'ShutterSpeedValue', 'ExifVersion'] + 'SceneCaptureType', 'SubjectDistance', 'CustomRendered', + 'SubjectDistanceRange', 'DigitalZoomRatio', 'ApertureValue', + 'ImageWidth', 'GPSDOP', 'MaxApertureValue', 'ColorSpace', + 'FocalLengthIn35mmFilm', 'ExposureMode', 'Saturation', 'ExposureTime', + 'ExifImageHeight', 'FNumber', 'YCbCrPositioning', 'Make', 'MeteringMode', + 'ExposureBiasValue', 'ExposureProgram', 'ComponentsConfiguration', + 'ExifImageWidth', 'ExifInteroperabilityOffset', 'BrightnessValue', + 'ImageLength', 'FlashPixVersion', 'SubsecTimeOriginal', 'Model', + 'SubsecTimeDigitized', 'ResolutionUnit', 'DateTimeOriginal', 'XResolution', + 'FocalLength', 'Sharpness', 'GPSLongitude', 'Contrast', 'Software', + 'GPSLatitude', 'MakerNote', 'GPSDateStamp', 'GPSAltitude', + 'GPSProcessingMethod', 'GPSTimeStamp', 'GPSLatitudeRef', 'WhiteBalance', + 'GPSLongitudeRef', 'Flash', 'SubsecTime', 'YResolution', + 'DateTimeDigitized', 'DateTime', 'GPSAltitudeRef', 'Orientation', + 'ShutterSpeedValue', 'ExifVersion'] + def validate_extracted_features(extracted_features, feature_names, expected_feature_names, data,