#!/usr/bin/python

# -*- coding: utf-8 -*-

import unittest

import cv

class Creating(unittest.TestCase):

        size=(640, 480)

        repeat=100

        def test_0_Create(self):

                image = cv.CreateImage(self.size, cv.IPL_DEPTH_8U, 1)

                cnt=cv.CountNonZero(image)

                self.assertEqual(cnt, 0, msg="Created image is not black. CountNonZero=%i" % cnt)

        def test_2_CreateRepeat(self):

                cnt=0

                for i in range(self.repeat):

                        image = cv.CreateImage(self.size, cv.IPL_DEPTH_8U, 1)

                        cnt+=cv.CountNonZero(image)

                self.assertEqual(cnt, 0, msg="Created images are not black. Mean CountNonZero=%.3f" % (1.*cnt/self.repeat))

        def test_2a_MemCreated(self):

                cnt=0

                v=[]

                for i in range(self.repeat):

                        image = cv.CreateImage(self.size, cv.IPL_DEPTH_8U, 1)

                        cv.FillPoly(image, [[(0, 0), (0, 100), (100, 0)]], 0)

                        cnt+=cv.CountNonZero(image)

                        v.append(image)

                self.assertEqual(cnt, 0, msg="Memorized images are not black. Mean CountNonZero=%.3f" % (1.*cnt/self.repeat))

        def test_3_tostirng(self):

                image = cv.CreateImage(self.size, cv.IPL_DEPTH_8U, 1)

                image.tostring()

                cnt=cv.CountNonZero(image)

                self.assertEqual(cnt, 0, msg="After tostring(): CountNonZero=%i" % cnt)

        def test_40_tostringRepeat(self):

                cnt=0

                image = cv.CreateImage(self.size, cv.IPL_DEPTH_8U, 1)

                for i in range(self.repeat*100):

                        image.tostring()

                cnt=cv.CountNonZero(image)

                self.assertEqual(cnt, 0, msg="Repeating tostring(): Mean CountNonZero=%.3f" % (1.*cnt/self.repeat))

        def test_41_CreateToStringRepeat(self):

                cnt=0

                for i in range(self.repeat*100):

                        image = cv.CreateImage(self.size, cv.IPL_DEPTH_8U, 1)

                        image.tostring()

                        cnt+=cv.CountNonZero(image)

                self.assertEqual(cnt, 0, msg="Repeating create and tostring(): Mean CountNonZero=%.3f" % (1.*cnt/self.repeat))

        def test_4a_MemCreatedToString(self):

                cnt=0

                v=[]

                for i in range(self.repeat):

                        image = cv.CreateImage(self.size, cv.IPL_DEPTH_8U, 1)

                        image.tostring()

                        cnt+=cv.CountNonZero(image)

                        v.append(image)

                self.assertEqual(cnt, 0, msg="Repeating and memorizing after tostring(): Mean CountNonZero=%.3f" % (1.*cnt/self.repeat))

if __name__ == "__main__":

    unittest.main()