Exemple #1
0
    def setUp(self):
        magnet1 = MagnetModel(1.0, 0.0, 1.5)
        magnet2 = MagnetModel(-1.0, 1.0, 1.5)
        magnet3 = MagnetModel(-1.0, -1.0, 1.5)
        magnet4 = MagnetModel(0.0, 0.0, 0.5)

        self.magnets = [magnet1, magnet2, magnet3, magnet4]

        self.plane_distance = 0.25
        self.friction = 0.3
        self.gravity_pullback = 0.5

        self.pendulum = PendulumModel(self.friction, self.gravity_pullback,
                                      self.plane_distance)
        self.pendulum.magnets = self.magnets

        self.time_step = 0.01
        self.integrator = EulerIntegrator(self.time_step)

        self.r = 255
        self.g = 0
        self.b = 0
        self.image_generator = BasicImageGenerator(self.r, self.g, self.b)

        self.size = 5
        self.resolution = 640
        self.basins_generator = BasinsGenerator(self.size, self.resolution)
        self.basins_generator.pendulum_model = self.pendulum
        self.basins_generator.integrator = self.integrator
        self.basins_generator.image_generator = self.image_generator
class TestBasicImageGenerator(unittest.TestCase):
    
    def setUp(self):
        self.r = 255
        self.g = 0
        self.b = 0
        self.image_generator = BasicImageGenerator(self.r, self.g, self.b)
    
    def test_generator_initalization(self):
        self.assertEqual(self.image_generator.base_hsv, colorsys.rgb_to_hsv(self.r / 255.0, self.g /255.0, self.b / 255.0))

    def test_image_generation_1(self):
        number_of_colors = 3
        test_data = numpy.array([[0, 2, 1], [1, 0, 1], [1, 2, 0]])

        self.image_generator.generate_image("test_image_1", test_data, test_data, number_of_colors)

        self.assertEqual(len(self.image_generator.color_list), number_of_colors)
       
    def test_image_generation_2(self): 
        number_of_colors = 10
        test_data = numpy.array([[random.randint(0, number_of_colors - 1) for j in range(1, 100)] for i in range(1, 100)])
                
        self.image_generator.generate_image("test_image_2", test_data, test_data, number_of_colors)    
        
        self.assertEqual(len(self.image_generator.color_list), number_of_colors)    

    def test_image_generation_3(self):
        number_of_colors = 5
        test_data = numpy.array([[0, 2, 2, 4, 1], [1, 0, 2, 4, 1], [1, 1, 0, 4, 1], [4, 4, 4, 0, 1], [3, 3, 3, 3, 0]])
        
        self.image_generator.generate_image("test_image_3", test_data, test_data, number_of_colors)
        
        self.assertEqual(len(self.image_generator.color_list), number_of_colors)
Exemple #3
0
class TestBasicImageGenerator(unittest.TestCase):
    def setUp(self):
        self.r = 255
        self.g = 0
        self.b = 0
        self.image_generator = BasicImageGenerator(self.r, self.g, self.b)

    def test_generator_initalization(self):
        self.assertEqual(
            self.image_generator.base_hsv,
            colorsys.rgb_to_hsv(self.r / 255.0, self.g / 255.0,
                                self.b / 255.0))

    def test_image_generation_1(self):
        number_of_colors = 3
        test_data = numpy.array([[0, 2, 1], [1, 0, 1], [1, 2, 0]])

        self.image_generator.generate_image("test_image_1", test_data,
                                            test_data, number_of_colors)

        self.assertEqual(len(self.image_generator.color_list),
                         number_of_colors)

    def test_image_generation_2(self):
        number_of_colors = 10
        test_data = numpy.array(
            [[random.randint(0, number_of_colors - 1) for j in range(1, 100)]
             for i in range(1, 100)])

        self.image_generator.generate_image("test_image_2", test_data,
                                            test_data, number_of_colors)

        self.assertEqual(len(self.image_generator.color_list),
                         number_of_colors)

    def test_image_generation_3(self):
        number_of_colors = 5
        test_data = numpy.array([[0, 2, 2, 4, 1], [1, 0, 2, 4, 1],
                                 [1, 1, 0, 4, 1], [4, 4, 4, 0, 1],
                                 [3, 3, 3, 3, 0]])

        self.image_generator.generate_image("test_image_3", test_data,
                                            test_data, number_of_colors)

        self.assertEqual(len(self.image_generator.color_list),
                         number_of_colors)
Exemple #4
0
 def setUp(self):
     self.r = 255
     self.g = 0
     self.b = 0
     self.image_generator = BasicImageGenerator(self.r, self.g, self.b)
 def setUp(self):
     self.r = 255
     self.g = 0
     self.b = 0
     self.image_generator = BasicImageGenerator(self.r, self.g, self.b)