def test_get_fextractor_subclasses(self): from core.noref_feature_extractor import NorefFeatureExtractor fextractor_subclasses = FeatureExtractor.get_subclasses_recursively() self.assertEquals(len(fextractor_subclasses), 7) self.assertTrue(VmafFeatureExtractor in fextractor_subclasses) self.assertTrue(MomentFeatureExtractor in fextractor_subclasses)
def test_get_fextractor_subclasses(self): fextractor_subclasses = FeatureExtractor.get_subclasses_recursively() self.assertEquals(len(fextractor_subclasses), 3) self.assertTrue(VmafFeatureExtractor in fextractor_subclasses) self.assertTrue(MomentFeatureExtractor in fextractor_subclasses)