def test_black(self):
gfx = GFX()
gfx.set_symbol('A1', GFXSymbol.black)
gfx.set_symbol('A2', GFXSymbol.black)
assert 'BLACK A1 A2' == gfx.output()
def test_white(self):
gfx = GFX()
gfx.set_symbol('A1', GFXSymbol.white)
gfx.set_symbol('A2', GFXSymbol.white)
assert 'WHITE A1 A2' == gfx.output()
def test_circle(self):
gfx = GFX()
gfx.set_symbol('A1', GFXSymbol.circle)
gfx.set_symbol('A2', GFXSymbol.circle)
assert 'CIRCLE A1 A2' == gfx.output()
def test_triangle(self):
gfx = GFX()
gfx.set_symbol('A1', GFXSymbol.triangle)
gfx.set_symbol('A2', GFXSymbol.triangle)
assert 'TRIANGLE A1 A2' == gfx.output()
def test_mark(self):
gfx = GFX()
gfx.set_symbol('A1', GFXSymbol.mark)
gfx.set_symbol('A2', GFXSymbol.mark)
assert 'MARK A1 A2' == gfx.output()
def test_square(self):
gfx = GFX()
gfx.set_symbol('A1', GFXSymbol.square)
gfx.set_symbol('A2', GFXSymbol.square)
assert 'SQUARE A1 A2' == gfx.output()