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'))
