from biokit.viz import Heatmap
from biokit.data.viz.heatmap_sample import load_heatmap_sample



def test_heatmap():
    data = load_heatmap_sample()
    h = Heatmap(data)
    h.plot(cmap='hot')
    h.row_method= 'single'
    h.col_method= 'single'
    #    category_cols=[0,0,1,1], 
    #    category_rows=[0,1,2,0,0,1,2,2,2,1])



@@ -39,115 +39,178 @@
 okit
-.data.viz.heatmap_sample import load_heatmap_sample%0A%0A%0A%0Adef test_heatmap():%0A    data = load_heatmap_sample()
+ import biokit_data%0A%0Adef test_heatmap():%0A    filename = biokit_data(%22test_heatmap.csv%22)%0A    import pandas as pd%0A    data = pd.read_csv(filename, skiprows=2, index_col=0)%0A
 %0A   

fix test