def test_feature_set_serialization(format, compressed): feature_set = FeatureSet( features=[ Features( recording_id='irrelevant', channels=0, start=0.0, duration=20.0, type='fbank', num_frames=2000, num_features=20, sampling_rate=16000, storage_type='lilcom', storage_path='/irrelevant/', storage_key='path.llc' ) ] ) with NamedTemporaryFile(suffix='.gz' if compressed else '') as f: if format == 'json': feature_set.to_json(f.name) feature_set_deserialized = FeatureSet.from_json(f.name) if format == 'yaml': feature_set.to_yaml(f.name) feature_set_deserialized = FeatureSet.from_yaml(f.name) assert feature_set_deserialized == feature_set
def test_feature_set_serialization(): feature_set = FeatureSet(feature_extractor=FeatureExtractor(), features=[ Features(recording_id='irrelevant', channel_id=0, start=0.0, duration=20.0, type='fbank', num_frames=2000, num_features=20, storage_type='lilcom', storage_path='/irrelevant/path.llc') ]) with NamedTemporaryFile() as f: feature_set.to_yaml(f.name) feature_set_deserialized = FeatureSet.from_yaml(f.name) assert feature_set_deserialized == feature_set