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
