Source

svgwrite / tests / test_defs.py

Full commit
#!/usr/bin/env python
#coding:utf-8
# Author:  mozman --<mozman@gmx.at>
# Purpose: test defs element
# Created: 25.09.2010
# Copyright (C) 2010, Manfred Moitzi
# License: MIT License

import sys
import unittest

from svgwrite.container import Group, Defs

class TestDefs(unittest.TestCase):
    def test_constructor(self):
        defs = Defs()
        self.assertEqual(defs.tostring(), "<defs />")

    def test_add_subelement(self):
        defs = Defs(id='defs')
        group = Group(id='group')
        defs.add(group)
        self.assertEqual(defs.tostring(), '<defs id="defs"><g id="group" /></defs>')

    def test_add_group(self):
        defs = Defs(id='defs')
        group = defs.add(Group(id='group')) # implicit call of add
        self.assertEqual(defs.tostring(), '<defs id="defs"><g id="group" /></defs>')

if __name__=='__main__':
    unittest.main()