def test_empty_section(self): section = EntitySection(Tags.from_text(EMPTYSEC), self.dwg) stream = StringIO() section.write(stream) result = stream.getvalue() stream.close() self.assertEqual(EMPTYSEC, result)
class TestEntitySection(unittest.TestCase): def setUp(self): self.dwg = DrawingProxy('AC1009') self.section = EntitySection(Tags.from_text(TESTENTITIES), self.dwg) def test_write(self): stream = StringIO() self.section.write(stream) result = stream.getvalue() stream.close() self.assertEqual(normlines(TESTENTITIES), normlines(result)) def test_empty_section(self): section = EntitySection(Tags.from_text(EMPTYSEC), self.dwg) stream = StringIO() section.write(stream) result = stream.getvalue() stream.close() self.assertEqual(EMPTYSEC, result) def test_iteration_with_layout_DXF12(self): dwg = ezdxf.new('AC1009') m = dwg.modelspace() m.add_line((0, 0), (1, 1)) entity = list(dwg.entities)[-1] self.assertEqual(dwg, entity.drawing) # check drawing attribute def test_iteration_with_layout_DXF2000(self): dwg = ezdxf.new('AC1015') m = dwg.modelspace() m.add_line((0, 0), (1, 1)) entity = list(dwg.entities)[-1] self.assertEqual(dwg, entity.drawing) # check drawing attribute def test_delete_all_entities_DXF12(self): dwg = ezdxf.new('AC1009') m = dwg.modelspace() for _ in range(5): m.add_line((0, 0), (1, 1)) self.assertEqual(5, len(dwg.entities)) dwg.entities.delete_all_entities() self.assertEqual(0, len(dwg.entities))
def setUp(self): self.dwg = DrawingProxy('AC1009') self.section = EntitySection(Tags.from_text(TESTENTITIES), self.dwg)