import unittest
from quad import Cell, build_quadtree

#from domain import *
from general_mesh import General_mesh as Mesh

import types, sys

#-------------------------------------------------------------

class TestCase(unittest.TestCase):

    def setUp(self):
        self.cell = Cell(100, 140, 0, 40, 'cell')
        
        a = [3, 107]
        b = [5, 107]
        c = [5, 105]
        d = [7, 107]
        e = [15, 115]
        f = [15, 130]
        g = [30, 110]
        h = [30, 130]

        points = [a, b, c, d, e, f, g, h]
        #bac, bce, ecf, dbe, daf, dae 
        vertices = [[1,0,2], [1,3,4], [1,2,3], [5,4,7], [4,6,7]] 

	mesh = Mesh(points, vertices)
        self.mesh = mesh
        Cell.initialise(mesh)
        
    def tearDown(self):
        pass

    def test_add_points_2_cell(self):
        self.cell.insert(0)
        self.cell.insert(1)
        
        result = self.cell.retrieve()
        assert type(result) in [types.ListType,types.TupleType],\
                                'should be a list'        
        self.assertEqual(len(result),2)
        
    def test_add_points_2_cellII(self):
        self.cell.insert([0,1,2,3,4,5,6,7])
        
        result = self.cell.retrieve()
        assert type(result) in [types.ListType,types.TupleType],\
	                        'should be a list'        
        self.assertEqual(len(result),8)


    def test_search(self):
        self.cell.insert([0,1,2,3,4,5,6,7])
	self.cell.split(4)
        
        result =  self.cell.search(x = 1, y = 101)
        assert type(result) in [types.ListType,types.TupleType],\
	                        'should be a list'        
        self.assertEqual(result, [0,1,2,3])
	
	
    def test_clear_1(self):
        self.cell.insert([0,1,2,3,4,5,6,7])
	assert self.cell.count() == 8	
	self.cell.clear()
	
        #This one actually revealed a bug :-)
	assert self.cell.count() == 0	
		
    def test_clear_2(self):
        self.cell.insert([0,1,2,3,4,5,6,7])
	assert self.cell.count() == 8	
	self.cell.split(2)
	assert self.cell.count() == 8		
		
	self.cell.clear()
	assert self.cell.count() == 0	
	
	
	
    def test_split(self):
        self.cell.insert([0,1,2,3,4,5,6,7], split = False)    
        
	#No children yet
	assert self.cell.children is None 
	assert self.cell.count() == 8

        #Split
	self.cell.split(4)
	#self.cell.show()
	#self.cell.show_all()
	
	
	#Now there are children
	assert self.cell.children is not None 
	assert self.cell.count() == 8	
	
	
		
    def test_collapse(self):
        self.cell.insert([0,1,2,3,4,5,6,7], split = False)    
        
        #Split maximally
	self.cell.split(1)
	
	#Now there are children
	assert self.cell.children is not None 
	assert self.cell.count() == 8	
	
	#Collapse
	self.cell.collapse(8)
	
	#No children
	assert self.cell.children is None 
	assert self.cell.count() == 8	

    def test_build_quadtree(self):

        Q = build_quadtree(self.mesh)
	assert Q.count() == 8

        #Q.show()
        
        result = Q.search(3, 105)
        assert type(result) in [types.ListType,types.TupleType],\
	                        'should be a list'
        
        self.assertEqual(result, [0,1,2,3])

        

#-------------------------------------------------------------
if __name__ == "__main__":

    mysuite = unittest.makeSuite(TestCase,'test')
    runner = unittest.TextTestRunner()
    runner.run(mysuite)