def test_generate_colors():
"Test functions for assigning colors to cells"
ds = datasets.get_uv()
cells_1 = ('A', 'B')
cells_2 = ('a', 'b', 'c')
colors = plot.colors_for_oneway(cells_1)
assert len(colors) == len(cells_1)
colors = plot.colors_for_twoway(cells_1, cells_2)
assert len(colors) == len(cells_1) * len(cells_2)
colors = plot.colors_for_twoway(cells_2, cells_1)
assert len(colors) == len(cells_1) * len(cells_2)
# colors_for_categorial()
f = ds['A']
colors = plot.colors_for_categorial(f)
for cell in f.cells:
assert cell in colors
i = ds.eval("A%B")
colors = plot.colors_for_categorial(i)
for cell in i.cells:
assert cell in colors
i = ds.eval("A%B%rm")
colors = plot.colors_for_categorial(i)
for cell in i.cells:
assert cell in colors
with pytest.raises(TypeError):
plot.colors_for_categorial("A%B")
def test_generate_colors():
"Test functions for assigning colors to cells"
ds = datasets.get_uv()
cells_1 = ('A', 'B')
cells_2 = ('a', 'b', 'c')
colors = plot.colors_for_oneway(cells_1)
assert_equal(len(colors), len(cells_1))
colors = plot.colors_for_twoway(cells_1, cells_2)
assert_equal(len(colors), len(cells_1) * len(cells_2))
colors = plot.colors_for_twoway(cells_2, cells_1)
assert_equal(len(colors), len(cells_1) * len(cells_2))
# colors_for_categorial()
f = ds['A']
colors = plot.colors_for_categorial(f)
for cell in f.cells:
assert_in(cell, colors)
i = ds.eval("A%B")
colors = plot.colors_for_categorial(i)
for cell in i.cells:
assert_in(cell, colors)
i = ds.eval("A%B%rm")
colors = plot.colors_for_categorial(i)
for cell in i.cells:
assert_in(cell, colors)
assert_raises(TypeError, plot.colors_for_categorial, "A%B")
def test_generate_colors():
"Test functions for assigning colors to cells"
ds = datasets.get_uv()
cells_1 = ('A', 'B')
cells_2 = ('a', 'b', 'c')
colors = plot.colors_for_oneway(cells_1)
eq_(len(colors), len(cells_1))
colors = plot.colors_for_twoway(cells_1, cells_2)
eq_(len(colors), len(cells_1) * len(cells_2))
colors = plot.colors_for_twoway(cells_2, cells_1)
eq_(len(colors), len(cells_1) * len(cells_2))
# colors_for_categorial()
f = ds['A']
colors = plot.colors_for_categorial(f)
for cell in f.cells:
assert_in(cell, colors)
i = ds.eval("A%B")
colors = plot.colors_for_categorial(i)
for cell in i.cells:
assert_in(cell, colors)
i = ds.eval("A%B%rm")
colors = plot.colors_for_categorial(i)
for cell in i.cells:
assert_in(cell, colors)
assert_raises(TypeError, plot.colors_for_categorial, "A%B")
def test_plot_colors():
"Test plotting color schemes"
cells_1 = ('A', 'B')
cells_2 = ('a', 'b', 'c')
colors = plot.colors_for_oneway(cells_1)
p = plot.ColorList(colors, show=False)
p.close()
colors = plot.colors_for_twoway(cells_1, cells_2)
p = plot.ColorList(colors, show=False)
p.close()
def test_plot_colors():
"Test plotting color schemes"
cells_1 = ('A', 'B')
cells_2 = ('a', 'b', 'c')
colors = plot.colors_for_oneway(cells_1)
p = plot.ColorList(colors, show=False)
p.close()
colors = plot.colors_for_twoway(cells_1, cells_2)
p = plot.ColorList(colors, show=False)
p.close()
def test_plot_colors():
"Test plotting color schemes"
cells_1 = ('A', 'B')
cells_2 = ('a', 'b', 'c')
colors = plot.colors_for_oneway(cells_1)
p = plot.ColorList(colors, show=False)
w0, h0 = p.figure.get_size_inches()
p.close()
p = plot.ColorList(colors, labels={'A': 'A' * 50, 'B': 'Bbb'}, show=False)
w, h = p.figure.get_size_inches()
eq_(h, h0)
assert_greater(w, w0)
p.close()
colors = plot.colors_for_twoway(cells_1, cells_2)
p = plot.ColorList(colors, show=False)
p.close()
def test_plot_colors():
"Test plotting color schemes"
cells_1 = ('A', 'B')
cells_2 = ('a', 'b', 'c')
colors = plot.colors_for_oneway(cells_1)
p = plot.ColorList(colors)
w0, h0 = p.figure.get_size_inches()
p.close()
p = plot.ColorList(colors, labels={'A': 'A'*50, 'B': 'Bbb'})
w, h = p.figure.get_size_inches()
assert h == h0
assert w > w0
p.close()
colors = plot.colors_for_twoway(cells_1, cells_2)
p = plot.ColorList(colors)
p.close()
def test_plot_colors():
"Test plotting color schemes"
cells_1 = ('A', 'B')
cells_2 = ('a', 'b', 'c')
colors = plot.colors_for_oneway(cells_1)
p = plot.ColorList(colors)
w0, h0 = p.figure.get_size_inches()
p.close()
p = plot.ColorList(colors, labels={'A': 'A'*50, 'B': 'Bbb'})
w, h = p.figure.get_size_inches()
eq_(h, h0)
assert_greater(w, w0)
p.close()
colors = plot.colors_for_twoway(cells_1, cells_2)
p = plot.ColorList(colors)
p.close()
def test_uts():
"test plot.UTS plotting function"
ds = datasets.get_uts()
x_long = set_tmin(concatenate(ds[:10, 'uts']), -1)
p = plot.UTS('uts', ds=ds)
p.close()
p = plot.UTS('uts', 'A%B', ds=ds)
p.set_ylim(1)
p.set_ylim(0, 1)
assert p.get_ylim() == (0, 1)
p.set_ylim(1, -1)
assert p.get_ylim() == (1, -1)
p.close()
p = plot.UTS(x_long, h=2, w=5, xlim=2)
assert p.get_xlim() == (-1, 1)
p.set_xlim(2, 4)
assert p.get_xlim() == (2, 4)
p.close()
# color dict
colors = plot.colors_for_oneway(['a0', 'a1', 'a2'])
a0, a1, a2 = ds[:3, 'uts']
a0.name = 'a0'
a1.name = 'a1'
a2.name = 'a2'
p = plot.UTS([[a0, a1, a2]], colors=colors)
p.close()
# multiple y with xax
y1 = ds.eval("uts[(A == 'a1') & (B == 'b1')]")
y1.name = 'y'
y2 = ds.eval("uts[(A == 'a0') & (B == 'b1')]")
y2.name = 'y2'
rm = ds.eval("rm[(A == 'a0') & (B == 'b1')]")
p = plot.UTS(y1, rm)
p.close()
p = plot.UTS([y1, y2], rm)
p.close()
