'''
Created on 12 dec. 2016

@author: olinox
'''
import unittest

from pypog import geometry, Grid, pencil

class Test(unittest.TestCase):

    def test_base_pencil(self):
        for cell_shape in (geometry.HEX, geometry.SQUARE):
        
            self.assertRaises(TypeError, pencil.BasePencil, "invalid arg")
            
            grid = Grid.Grid(cell_shape, 30, 30)
            my_pencil = pencil.BasePencil(grid)
            
            # default origin and position
            self.assertEqual(my_pencil.origin, None)
            self.assertEqual(my_pencil.position, None)
            self.assertRaises(AttributeError, my_pencil.origin, (1,1))
            self.assertRaises(AttributeError, my_pencil.position, (1,1))
            
            # size
            self.assertRaises(TypeError, setattr, my_pencil, "size", "a")
            self.assertRaises(ValueError, setattr, my_pencil, "size", -1)
            self.assertEqual(my_pencil.size, 1)
            
            # selection, added, removed
            self.assertEqual(my_pencil.selection, [])
            self.assertEqual(my_pencil.added, [])
            self.assertEqual(my_pencil.removed, [])
            
            # pencil methods
            self.assertRaises(TypeError, my_pencil.start, "a")
            self.assertRaises(pencil.NotStartedException, my_pencil.update, 1,1)
            self.assertRaises(NotImplementedError, my_pencil._update)
            
            try:
                my_pencil.start(0,0)
            except NotImplementedError:
                pass
            self.assertRaises(TypeError, my_pencil.update, "a")
            self.assertEqual(my_pencil.origin, (0,0))
            
    def test_line_pencil(self):
        pass
    
    def test_free_pencil(self):
        pass
    
    def test_pot_pencil(self):
        pass
    
    def test_rect_pencil(self):
        pass
    
    def test_hrect_pencil(self):
        pass
    
    def test_boundary_pencil(self):
        pass
    
    

if __name__ == "__main__":
    unittest.main()