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