def index(): form = MushroomFeaturesForm() result = None if form.validate_on_submit(): tree_data = grab_tree(current_app.config['TREE_DATA_FILE_NAME']) data = [ form.cap_shape.data, form.cap_surface.data, form.cap_color.data, form.bruises.data, form.odor.data, form.gill_attachment.data, form.gill_spacing.data, form.gill_size.data, form.gill_color.data, form.stalk_shape.data, form.stalk_root.data, form.stalk_surface_above_ring.data, form.stalk_surface_below_ring.data, form.stalk_color_above_ring.data, form.stalk_color_below_ring.data, form.veil_type.data, form.veil_color.data, form.ring_number.data, form.ring_type.data, form.spore_print_color.data, form.population.data, form.habitat.data ] result = classify(tree_data, DATASET_LABELS, data) return render_template('main/index.html', form=form, result=result)
def test_classify(self): self.assertEqual( classify( self.tree_data, DATASET_LABELS, 'x,s,g,f,n,f,w,b,k,t,e,s,s,w,w,p,w,o,e,n,a,g'.split(',') ), 'e' ) self.assertEqual( classify( self.tree_data, DATASET_LABELS, 'x,y,n,t,p,f,c,n,w,e,e,s,s,w,w,p,w,o,p,n,s,u'.split(',') ), 'p' ) self.assertEqual( classify( self.tree_data, DATASET_LABELS, 'f,f,g,f,n,f,w,b,n,t,e,f,f,w,w,p,w,o,e,n,a,g'.split(',') ), 'e' ) self.assertEqual( classify( self.tree_data, DATASET_LABELS, 'f,f,n,f,n,f,w,b,h,t,e,s,s,w,w,p,w,o,e,n,a,g'.split(',') ), 'e' ) self.assertEqual( classify( self.tree_data, DATASET_LABELS, 'x,f,g,f,f,f,c,b,p,e,b,k,k,b,b,p,w,o,l,h,v,p'.split(',') ), 'p' ) self.assertEqual( classify( self.tree_data, DATASET_LABELS, 'x,y,g,f,f,f,c,b,p,e,b,k,k,b,b,p,w,o,l,h,y,g'.split(',') ), 'p' )