def test_subset(large_prosodylab_format_directory, temp_dir, large_dataset_dictionary):
output_directory = os.path.join(temp_dir, 'large_subset')
shutil.rmtree(output_directory, ignore_errors=True)
d = Dictionary(large_dataset_dictionary, output_directory)
d.write()
c = Corpus(large_prosodylab_format_directory, output_directory)
c.initialize_corpus(d)
sd = c.split_directory()
fc = FeatureConfig()
fc.generate_features(c)
s = c.subset_directory(10, fc)
assert os.path.exists(sd)
assert os.path.exists(s)
def test_subset(large_prosodylab_format_directory, temp_dir, large_dataset_dictionary):
output_directory = os.path.join(temp_dir, 'large_subset')
shutil.rmtree(output_directory, ignore_errors=True)
d = Dictionary(large_dataset_dictionary, output_directory)
d.write()
c = Corpus(large_prosodylab_format_directory, output_directory)
c.initialize_corpus(d)
sd = c.split_directory()
fc = FeatureConfig()
fc.generate_features(c)
s = c.subset_directory(10, fc)
assert os.path.exists(sd)
assert os.path.exists(s)
