source: trunk/misc/tools/acceptance_tests/mandelbrot/test_calculate_point.py @ 8427

"""
  Test of calculate_point within mandelbrot.py
"""

import unittest
import Numeric
from mandelbrot import calculate_point
#from mandel_ext import calculate_point


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

class TestCase(unittest.TestCase):

    def setUp(self):
        pass

        
    def test_basic(self):
        """Test that c==0 always returns kmax
        """

        c = complex(0, 0)
        
        for kmax in [0,1,2,3,4,10, 20, 33, 45, 100]:
            count = calculate_point(c, kmax)    
            assert count == kmax        

            
    def test_points_clearly_outside_set(self):  
    
        for kmax in [1,2,3,5,10,20,100]:
            for c in [complex(5, 4), complex(2,1), complex(-2,1),
                      complex(2,-1), complex(-2,-1)]:
                count = calculate_point(c, kmax)
                assert count == 1
        
                
    def test_arbitrary_points(self):
        for kmax in [10,20,100]:            
            c = complex(1, 1.5)
            count = calculate_point(c, kmax)
            assert count == 2    
            
            c = complex(-1, 1.5)
            count = calculate_point(c, kmax)
            assert count == 2               
            
            c = complex(-1, -1.5)
            count = calculate_point(c, kmax)
            assert count == 2                       
            
            c = complex(1, -1.5)
            count = calculate_point(c, kmax)
            assert count == 2                               
            
            c = complex(1, 0)
            count = calculate_point(c, kmax)
            assert count == 3                                       
            
            c = complex(0.2, 1)
            count = calculate_point(c, kmax)
            assert count == 4                                               
            
            c = complex(0.2, 0.7)
            count = calculate_point(c, kmax)
            assert count == 6                                                       
            
            c = complex(-0.2, 0.9)
            count = calculate_point(c, kmax)
            assert count == 9
            

    def test_another_known_point(self):
        c = complex(0.5, 0.5)
        count = calculate_point(c, 256)
        assert count == 5        

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

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