def test_extract_at_plant_scale():
files = shared_data(openalea.strawberry).glob('*.mtg')
mtg_path = dict((name(f), f) for f in files)
mtg = MTG()
for k, genotype in enumerate(mtg_path):
if (genotype=="Sicile") or (genotype=="Nils") or (genotype=='origine_stolon_gariguetteDV_6novembre2018') \
or (genotype=='friendlyfruit_varieties'):
pass
else:
mtg = union(mtg, read_mtg_file(mtg_path[genotype]))
genotypes = mtg.property('Genotype')
for geno in set(genotypes.values()):
vids = [
vid for vid in mtg.vertices(scale=1) if genotypes.get(vid) == geno
]
df = extract_at_plant_scale(mtg, vids)
assert not df.empty
assert set(df['Genotype']) == {geno}
