def test_check_if_extracted_path():
path = "./tests/data/features"
feature_extractor = FeatureExtractor()
feature_extractor.set_as_extracted(path)
json_features = os.path.join(path, "parameters.json")
_clean(json_features)
feature_extractor.set_as_extracted(path)
assert feature_extractor.check_if_extracted_path(path)
def test_check_if_extracted():
dataset_path = "./tests/data"
dataset = Dataset(dataset_path)
path = "./tests/data/features/FeatureExtractor/original"
mkdir_if_not_exists(path, parents=True)
feature_extractor = FeatureExtractor()
json_features = os.path.join(path, "parameters.json")
_clean(json_features)
feature_extractor.set_as_extracted(path)
assert feature_extractor.check_if_extracted(dataset)
def test_set_as_extracted():
path = "./tests/data/features"
feature_extractor = FeatureExtractor()
feature_extractor.set_as_extracted(path)
json_features = os.path.join(path, "parameters.json")
assert os.path.exists(json_features)
parameters_features = load_json(json_features)
assert len(parameters_features) == 8
default_parameters = {
"sequence_time": 1.0,
"sequence_hop_time": 0.5,
"audio_hop": 680,
"audio_win": 1024,
"sr": 22050,
"sequence_frames": 32,
"sequence_hop": 16,
"features_folder": "features",
}
for key in default_parameters.keys():
assert default_parameters[key] == parameters_features[key]
def test_set_as_extracted():
path = './tests/data/features'
feature_extractor = FeatureExtractor()
feature_extractor.set_as_extracted(path)
json_features = os.path.join(path, "parameters.json")
assert os.path.exists(json_features)
parameters_features = load_json(json_features)
assert len(parameters_features) == 8
default_parameters = {
'sequence_time': 1.0,
'sequence_hop_time': 0.5,
'audio_hop': 680,
'audio_win': 1024,
'sr': 22050,
'sequence_frames': 32,
'sequence_hop': 16,
'features_folder': 'features'
}
for key in default_parameters.keys():
assert default_parameters[key] == parameters_features[key]