File: test_image.py

package info (click to toggle)
pigment-python 0.3.4-2
  • links: PTS
  • area: main
  • in suites: lenny
  • size: 2,856 kB
  • ctags: 2,881
  • sloc: python: 11,567; sh: 9,133; makefile: 227; ansic: 76
file content (96 lines) | stat: -rw-r--r-- 3,173 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
# -*- mode: python; coding: utf-8 -*-
#
# Pigment Python binding unit tests
#
# Copyright © 2006, 2007, 2008 Fluendo Embedded S.L.
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the
# Free Software Foundation, Inc., 59 Temple Place - Suite 330,
# Boston, MA 02111-1307, USA.

from test_drawable import _TestDrawable
import pgm
import gst

class TestImage(_TestDrawable):

    def setUp(self):
        self.drawable = pgm.Image()
        _TestDrawable.setUp(self)

    def test_interp(self):

        # get
        interp = self.drawable.get_interp()
        self.assertEqual(interp, pgm.ImageInterpType(pgm.IMAGE_BILINEAR))

        # set
        new_interp = pgm.ImageInterpType(pgm.IMAGE_NEAREST)
        self.drawable.set_interp(new_interp)
        self.assertEqual(self.drawable.get_interp(), new_interp)

    def test_aspect_ratio(self):

        # get
        par = self.drawable.get_aspect_ratio()
        self.failUnless(isinstance(par, tuple))
        self.assertEqual(len(par), 2)
        self.assertEqual(par, (0, 1))

        # set
        new_par = (4, 3)
        self.drawable.set_aspect_ratio(*new_par)
        self.assertEqual(self.drawable.get_aspect_ratio(), new_par)

##     def test_set_from(self):

##         # gst buffer
##         buf = gst.Buffer("foo")
##         err = self.drawable.set_from_gst_buffer(pgm.ImagePixelFormat(pgm.IMAGE_RGBA),
##                                                 800, 600, 1, buf)
##         self.assertEqual(err, pgm.Error(pgm.ERROR_OK))
##         self.assertEqual(buf.__grefcount__, 2)

##         # buffer
##         buf = "foo"
##         err = self.drawable.set_from_buffer(pgm.ImagePixelFormat(pgm.IMAGE_RGBA),
##                                             800, 600, 1, len(buf), buf)
##         self.assertEqual(err, pgm.Error(pgm.ERROR_OK))


    def test_clear(self):

        # gst buf ref count
        buf = gst.Buffer("foo")
        err = self.drawable.set_from_gst_buffer(pgm.ImagePixelFormat(pgm.IMAGE_RGBA),
                                                800, 600, 1, buf)
        self.assertEqual(err, pgm.Error(pgm.ERROR_OK))
        self.assertEqual(buf.__grefcount__, 2)

        self.drawable.clear()
        self.assertEqual(buf.__grefcount__, 1)

        # TODO: slave

    def test_layout(self):

        # get
        ltype = self.drawable.get_layout()
        self.assertEqual(ltype, pgm.ImageLayoutType(pgm.IMAGE_SCALED))

        # set
        new_ltype = pgm.ImageLayoutType(pgm.IMAGE_FILLED)
        self.drawable.set_layout(new_ltype)
        self.assertEqual(self.drawable.get_layout(), new_ltype)