test_weighted_grid.py

#!/usr/bin/env python

import os, sys
import unittest

sys.path.append('../../')

from myplotlib import EmptyFig
from myplotlib.mpl_toolbox import WeightedGrid, Ax2dPlots

class WeightedGridTestCase(unittest.TestCase):

    def setUp(self):
        self.rawdata = [-1., 1.]
        self.foo_fig = EmptyFig()
        self.foo_grid = WeightedGrid()
        self.foo_grid2 = WeightedGrid()
        self.foo_axes = Ax2dPlots(self.foo_fig)
        self.foo_axes2 = Ax2dPlots(self.foo_fig)

    def test_default_constructor(self):
        wgrid = WeightedGrid()
        self.assertEqual(wgrid.weights, [])

    def test_constructor_with_weights(self):
        wgrid = WeightedGrid(weights=[0.2, 0.2, 0.4, 0.15, 0.05])
        self.assertEqual(wgrid.weights, [0.2, 0.2, 0.4, 0.15, 0.05])
        
    def test_append_axes(self):
        wgrid = WeightedGrid(margin=0.15, padding=0.05)
        self.foo_fig.set_grid(wgrid)
        wgrid.append_axes(self.foo_axes, 'foo_axes', self.rawdata)
        self.assertEqual(wgrid.weights, [1.])
        wgrid.append_axes(self.foo_axes2, 'foo_axes2', self.rawdata)
        self.assertEqual(wgrid.weights, [1., 1.])

    def test_append_axes_with_weight(self):
        wgrid = WeightedGrid(margin=0.15, padding=0.05)
        self.foo_fig.set_grid(wgrid)
        wgrid.append_axes(self.foo_axes, 'foo_axes', self.rawdata, weight=0.6)
        self.assertEqual(wgrid.weights, [0.6])
        wgrid.append_axes(self.foo_axes2, 'foo_axes2', self.rawdata, weight=0.4)
        self.assertEqual(wgrid.weights, [0.6, 0.4])
        
    def test_append_axes_with_weights(self):
        wgrid = WeightedGrid(margin=0.15, padding=0.05)
        self.foo_fig.set_grid(wgrid)
        wgrid.append_axes(self.foo_axes, 'foo_axes', self.rawdata)
        self.assertEqual(wgrid.weights, [1.])
        wgrid.append_axes(self.foo_axes2, 'foo_axes2', self.rawdata, weights=[0.4, 0.6])
        self.assertEqual(wgrid.weights, [0.4, 0.6])

    def test_append_grid(self):
        wgrid = WeightedGrid(margin=0.15, padding=0.05)
        self.foo_fig.set_grid(wgrid)
        wgrid.append_grid(self.foo_grid)
        self.assertEqual(wgrid.weights, [1.])
        wgrid.append_grid(self.foo_grid2)
        self.assertEqual(wgrid.weights, [1., 1.])

    def test_append_grid_with_weight(self):
        wgrid = WeightedGrid(margin=0.15, padding=0.05)
        self.foo_fig.set_grid(wgrid)
        wgrid.append_grid(self.foo_grid, weight=0.6)
        self.assertEqual(wgrid.weights, [0.6])
        wgrid.append_grid(self.foo_grid2, weight=0.4)
        self.assertEqual(wgrid.weights, [0.6, 0.4])        

    def test_append_grid_with_weights(self):
        wgrid = WeightedGrid(margin=0.15, padding=0.05)
        self.foo_fig.set_grid(wgrid)
        wgrid.append_grid(self.foo_grid)
        self.assertEqual(wgrid.weights, [1.])
        wgrid.append_grid(self.foo_grid2, weights=[0.4, 0.6])
        self.assertEqual(wgrid.weights, [0.4, 0.6])        

    def test_get_item_weight(self):
        wgrid = WeightedGrid(margin=0.15, padding=0.05, weights=[0.5, 0.3, 0.6])
        self.assertEqual(wgrid.get_item_weight(1), 0.3)

    def test_get_weights(self):
        wgrid = WeightedGrid(margin=0.15, padding=0.05, weights=[0.5, 0.3, 0.6])
        self.assertEqual(wgrid.get_weights(), [0.5, 0.3, 0.6])

    def test_looping_grids(self):
        wgrid = WeightedGrid(margin=0.15, padding=0.05)
        wgrid2 = WeightedGrid(margin=0.15, padding=0.05)

        self.foo_fig.set_grid(wgrid)
        
        wgrid.append_grid(wgrid2)
        wgrid2.append_grid(wgrid)

        with self.assertRaises(RuntimeError):
            self.foo_fig.layout()      
        
class HorizontalWeightedGridTestCase(unittest.TestCase):

    def setUp(self):
        self.rawdata = [-1., 1.]
        self.foo_fig = EmptyFig()
        self.foo_grid = HorizontalWeightedGrid()
        self.foo_axes = Ax2dPlots(self.foo_fig)
        
        self.wgrid = HorizontalWeightedGrid(margin=0.15, padding=0.05)
        
        self.foo_fig.set_grid(wgrid)
        
        self.wgrid.append_axes(self.foo_axes, 'foo_axes', self.rawdata, weight=0.6)
        self.wgrid.append_axes(self.foo_grid, weight=0.4)

    def test_layout(self):
        self.foo_fig.layout()
    
class VerticalWeightedGridTestCase(unittest.TestCase):

    def setUp(self):
        self.rawdata = [-1., 1.]
        self.foo_fig = EmptyFig()
        self.foo_grid = VerticalWeightedGrid()
        self.foo_axes = Ax2dPlots(self.foo_fig)
        
        self.wgrid = VerticalWeightedGrid(margin=0.15, padding=0.05)
        
        self.foo_fig.set_grid(wgrid)
        
        self.wgrid.append_axes(self.foo_axes, 'foo_axes', self.rawdata, weight=0.6)
        self.wgrid.append_axes(self.foo_grid, weight=0.4)

    def test_layout(self):
        self.foo_fig.layout()
        
if __name__ == '__main__':
    unittest.main()