def test_get_color(): eq_(view._get_color(None, NeuriteType.basal_dendrite), "red") eq_(view._get_color(None, NeuriteType.axon), "blue") eq_(view._get_color(None, NeuriteType.apical_dendrite), "purple") eq_(view._get_color(None, NeuriteType.soma), "black") eq_(view._get_color(None, NeuriteType.undefined), "green") eq_(view._get_color(None, 'wrong'), "green") eq_(view._get_color('blue', 'wrong'), "blue") eq_(view._get_color('yellow', NeuriteType.axon), "yellow")
def test_get_color(): nt.eq_(view._get_color(None, NeuriteType.basal_dendrite), "red") nt.eq_(view._get_color(None, NeuriteType.axon), "blue") nt.eq_(view._get_color(None, NeuriteType.apical_dendrite), "purple") nt.eq_(view._get_color(None, NeuriteType.soma), "black") nt.eq_(view._get_color(None, NeuriteType.undefined), "green") nt.eq_(view._get_color(None, 'wrong'), "green") nt.eq_(view._get_color('blue', 'wrong'), "blue") nt.eq_(view._get_color('yellow', NeuriteType.axon), "yellow")
def test_get_color(): assert view._get_color(None, NeuriteType.basal_dendrite) == "red" assert view._get_color(None, NeuriteType.axon) == "blue" assert view._get_color(None, NeuriteType.apical_dendrite) == "purple" assert view._get_color(None, NeuriteType.soma) == "black" assert view._get_color(None, NeuriteType.undefined) == "green" assert view._get_color(None, 'wrong') == "green" assert view._get_color('blue', 'wrong') == "blue" assert view._get_color('yellow', NeuriteType.axon) == "yellow"