def test_t_contrast_parsing(): "Test parsing of t-contrast expressions" y = np.arange(9.).reshape((3, 3)) indexes = {'a': 0, 'b': 1, 'c': 2} contrast = "sum(a>c, b>c)" contrast_ = t_contrast.parse(contrast) eq_(contrast_, ('afunc', np.sum, (('comp', 'a', 'c'), ('comp', 'b', 'c')))) _, cells = t_contrast._t_contrast_rel_properties(contrast_) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, ('a', 'b', 'c')) data = t_contrast._t_contrast_rel_data(y, indexes, pc, mc) assert_equal(data['a'], np.arange(0., 3.)) assert_equal(data['b'], np.arange(3., 6.)) assert_equal(data['c'], np.arange(6., 9.)) contrast = "sum(a>*, b>*)" contrast_ = t_contrast.parse(contrast) eq_(contrast_, ('afunc', np.sum, (('comp', 'a', '*'), ('comp', 'b', '*')))) _, cells = t_contrast._t_contrast_rel_properties(contrast_) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, ('a', 'b', 'c')) data = t_contrast._t_contrast_rel_data(y, indexes, pc, mc) assert_equal(data['a'], np.arange(0., 3.)) assert_equal(data['b'], np.arange(3., 6.)) assert_equal(data['*'], y.mean(0)) assert_raises(ValueError, t_contrast._t_contrast_rel_expand_cells, cells, ('a|c', 'b|c', 'c|c')) # test finding cells all_cells = (('fondue pot', 'brie'), ('fondue mix', 'brie'), ('fondue pot', 'edam'), ('raclette', 'edam')) cells = (('* pot', '*'), ('fondue *', 'brie')) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, all_cells) eq_(pc, set(all_cells[:3])) eq_( mc, { ('* pot', '*'): (('fondue pot', 'brie'), ('fondue pot', 'edam')), ('fondue *', 'brie'): (('fondue pot', 'brie'), ('fondue mix', 'brie')) })
def test_t_contrast_parsing(): "Test parsing of t-contrast expressions" y = np.arange(9.0).reshape((3, 3)) indexes = {"a": 0, "b": 1, "c": 2} contrast = "sum(a>c, b>c)" contrast_ = t_contrast.parse(contrast) eq_(contrast_, ("afunc", np.sum, (("comp", "a", "c"), ("comp", "b", "c")))) _, cells = t_contrast._t_contrast_rel_properties(contrast_) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, ("a", "b", "c")) data = t_contrast._t_contrast_rel_data(y, indexes, pc, mc) assert_equal(data["a"], np.arange(0.0, 3.0)) assert_equal(data["b"], np.arange(3.0, 6.0)) assert_equal(data["c"], np.arange(6.0, 9.0)) contrast = "sum(a>*, b>*)" contrast_ = t_contrast.parse(contrast) eq_(contrast_, ("afunc", np.sum, (("comp", "a", "*"), ("comp", "b", "*")))) _, cells = t_contrast._t_contrast_rel_properties(contrast_) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, ("a", "b", "c")) data = t_contrast._t_contrast_rel_data(y, indexes, pc, mc) assert_equal(data["a"], np.arange(0.0, 3.0)) assert_equal(data["b"], np.arange(3.0, 6.0)) assert_equal(data["*"], y.mean(0)) assert_raises(ValueError, t_contrast._t_contrast_rel_expand_cells, cells, ("a|c", "b|c", "c|c")) # test finding cells all_cells = (("fondue pot", "brie"), ("fondue mix", "brie"), ("fondue pot", "edam"), ("raclette", "edam")) cells = (("* pot", "*"), ("fondue *", "brie")) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, all_cells) eq_(pc, set(all_cells[:3])) eq_( mc, { ("* pot", "*"): (("fondue pot", "brie"), ("fondue pot", "edam")), ("fondue *", "brie"): (("fondue pot", "brie"), ("fondue mix", "brie")), }, )
def test_t_contrast_parsing(): "Test parsing of t-contrast expressions" y = np.arange(9.).reshape((3, 3)) indexes = {'a': 0, 'b': 1, 'c': 2} contrast = "sum(a>c, b>c)" contrast_ = t_contrast.parse(contrast) eq_(contrast_, ('afunc', np.sum, (('comp', 'a', 'c'), ('comp', 'b', 'c')))) _, cells = t_contrast._t_contrast_rel_properties(contrast_) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, ('a', 'b', 'c')) data = t_contrast._t_contrast_rel_data(y, indexes, pc, mc) assert_equal(data['a'], np.arange(0., 3.)) assert_equal(data['b'], np.arange(3., 6.)) assert_equal(data['c'], np.arange(6., 9.)) contrast = "sum(a>*, b>*)" contrast_ = t_contrast.parse(contrast) eq_(contrast_, ('afunc', np.sum, (('comp', 'a', '*'), ('comp', 'b', '*')))) _, cells = t_contrast._t_contrast_rel_properties(contrast_) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, ('a', 'b', 'c')) data = t_contrast._t_contrast_rel_data(y, indexes, pc, mc) assert_equal(data['a'], np.arange(0., 3.)) assert_equal(data['b'], np.arange(3., 6.)) assert_equal(data['*'], y.mean(0)) assert_raises(ValueError, t_contrast._t_contrast_rel_expand_cells, cells, ('a|c', 'b|c', 'c|c')) # test finding cells all_cells = (('fondue pot', 'brie'), ('fondue mix', 'brie'), ('fondue pot', 'edam'), ('raclette', 'edam')) cells = (('* pot', '*'), ('fondue *', 'brie')) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, all_cells) eq_(pc, set(all_cells[:3])) eq_(mc, {('* pot', '*'): (('fondue pot', 'brie'), ('fondue pot', 'edam')), ('fondue *', 'brie'): (('fondue pot', 'brie'), ('fondue mix', 'brie'))})
def test_t_contrast_parsing(): "Test parsing of t-contrast expressions" y = np.arange(9.).reshape((3, 3)) indexes = {'a': 0, 'b': 1, 'c': 2} contrast = "sum(a>c, b>c)" contrast_ = t_contrast._parse_t_contrast(contrast) eq_(contrast_, ('afunc', np.sum, [('comp', 'a', 'c'), ('comp', 'b', 'c')])) contrast = "sum(a>*, b>*)" contrast_ = t_contrast._parse_t_contrast(contrast) eq_(contrast_, ('afunc', np.sum, [('comp', 'a', '*'), ('comp', 'b', '*')])) _, cells = t_contrast._t_contrast_rel_properties(contrast_) pc, mc = t_contrast._t_contrast_rel_expand_cells(cells, ('a', 'b', 'c')) data = t_contrast._t_contrast_rel_data(y, indexes, pc, mc) assert_equal(data['a'], np.arange(3.)) assert_equal(data['*'], y.mean(0)) assert_raises(ValueError, t_contrast._t_contrast_rel_expand_cells, cells, ('a|c', 'b|c', 'c|c'))