def test_empty_section(self):
section = ClassesSection(Tags.from_text(EMPTYSEC), self.dwg)
stream = StringIO()
section.write(stream)
result = stream.getvalue()
stream.close()
self.assertEqual(EMPTYSEC, result)
def test_empty_section():
tags = load_section(EMPTYSEC, 'CLASSES')
section = ClassesSection(tags, None)
stream = StringIO()
section.write(TagWriter(stream))
result = stream.getvalue()
stream.close()
assert EMPTYSEC == result
class TestClassesSection(unittest.TestCase):
def setUp(self):
self.dwg = DrawingProxy('AC1009')
self.section = ClassesSection(Tags.from_text(TESTCLASSES), self.dwg)
def test_write(self):
stream = StringIO()
self.section.write(stream)
result = stream.getvalue()
stream.close()
self.assertEqual(normlines(TESTCLASSES), normlines(result))
def test_empty_section(self):
section = ClassesSection(Tags.from_text(EMPTYSEC), self.dwg)
stream = StringIO()
section.write(stream)
result = stream.getvalue()
stream.close()
self.assertEqual(EMPTYSEC, result)