import pytest
from unittest import TestCase

from stylo.shape import Ellipse, Circle, Rectangle, Square
from stylo.testing.shape import BaseShapeTest


@pytest.mark.shape
class TestEllipse(TestCase, BaseShapeTest):
    """Tests for the :code:`Ellipse` shape."""

    def setUp(self):
        self.shape = Ellipse(0, 0, 1 / 2, 1 / 3, 0.6)


@pytest.mark.shape
class TestCircle(TestCase, BaseShapeTest):
    """Tests for the :code:`Circle` shape."""

    def setUp(self):
        self.shape = Circle(0, 0, 0.5)


@pytest.mark.shape
class TestRectangle(TestCase, BaseShapeTest):
    """Tests for the :code:`Rectangle` shape."""

    def setUp(self):
        self.shape = Rectangle(0, 0, 0.6, 0.3)


@pytest.mark.shape
class TestSquare(TestCase, BaseShapeTest):
    """Tests for the :code:`Square` shape."""

    def setUp(self):
        self.shape = Square(0, 0, 0.75)

import pytest
from unittest import TestCase

from stylo.shape import Ellipse, Circle, Rectangle, Square, Triangle
from stylo.testing.shape import BaseShapeTest


@pytest.mark.shape
class TestEllipse(TestCase, BaseShapeTest):
    """Tests for the :code:`Ellipse` shape."""

    def setUp(self):
        self.shape = Ellipse(0, 0, 1 / 2, 1 / 3, 0.6)


@pytest.mark.shape
class TestCircle(TestCase, BaseShapeTest):
    """Tests for the :code:`Circle` shape."""

    def setUp(self):
        self.shape = Circle(0, 0, 0.5)


@pytest.mark.shape
class TestRectangle(TestCase, BaseShapeTest):
    """Tests for the :code:`Rectangle` shape."""

    def setUp(self):
        self.shape = Rectangle(0, 0, 0.6, 0.3)


@pytest.mark.shape
class TestSquare(TestCase, BaseShapeTest):
    """Tests for the :code:`Square` shape."""

    def setUp(self):
        self.shape = Square(0, 0, 0.75)

@pytest.mark.shape
class TestTriangle(TestCase, BaseShapeTest):
    """Tests for the :code:`Triangle` shape."""

    def setUp(self):
        self.shape = Triangle((1,.5),(.2,1),(.4,.5))

Add Triangle to shape tests
Add Triangle to shape tests

  import pytest
  from unittest import TestCase
  
- from stylo.shape import Ellipse, Circle, Rectangle, Square
+ from stylo.shape import Ellipse, Circle, Rectangle, Square, Triangle
  from stylo.testing.shape import BaseShapeTest
  
  
  @pytest.mark.shape
  class TestEllipse(TestCase, BaseShapeTest):
      """Tests for the :code:`Ellipse` shape."""
  
      def setUp(self):
          self.shape = Ellipse(0, 0, 1 / 2, 1 / 3, 0.6)
  
  
  @pytest.mark.shape
  class TestCircle(TestCase, BaseShapeTest):
      """Tests for the :code:`Circle` shape."""
  
      def setUp(self):
          self.shape = Circle(0, 0, 0.5)
  
  
  @pytest.mark.shape
  class TestRectangle(TestCase, BaseShapeTest):
      """Tests for the :code:`Rectangle` shape."""
  
      def setUp(self):
          self.shape = Rectangle(0, 0, 0.6, 0.3)
  
  
  @pytest.mark.shape
  class TestSquare(TestCase, BaseShapeTest):
      """Tests for the :code:`Square` shape."""
  
      def setUp(self):
          self.shape = Square(0, 0, 0.75)
  
+ @pytest.mark.shape
+ class TestTriangle(TestCase, BaseShapeTest):
+     """Tests for the :code:`Triangle` shape."""
+ 
+     def setUp(self):
+         self.shape = Triangle((1,.5),(.2,1),(.4,.5))
+ 
Add Triangle to shape tests
## Code Before:
import pytest
from unittest import TestCase

from stylo.shape import Ellipse, Circle, Rectangle, Square
from stylo.testing.shape import BaseShapeTest


@pytest.mark.shape
class TestEllipse(TestCase, BaseShapeTest):
    """Tests for the :code:`Ellipse` shape."""

    def setUp(self):
        self.shape = Ellipse(0, 0, 1 / 2, 1 / 3, 0.6)


@pytest.mark.shape
class TestCircle(TestCase, BaseShapeTest):
    """Tests for the :code:`Circle` shape."""

    def setUp(self):
        self.shape = Circle(0, 0, 0.5)


@pytest.mark.shape
class TestRectangle(TestCase, BaseShapeTest):
    """Tests for the :code:`Rectangle` shape."""

    def setUp(self):
        self.shape = Rectangle(0, 0, 0.6, 0.3)


@pytest.mark.shape
class TestSquare(TestCase, BaseShapeTest):
    """Tests for the :code:`Square` shape."""

    def setUp(self):
        self.shape = Square(0, 0, 0.75)

## Instruction:
Add Triangle to shape tests
## Code After:
import pytest
from unittest import TestCase

from stylo.shape import Ellipse, Circle, Rectangle, Square, Triangle
from stylo.testing.shape import BaseShapeTest


@pytest.mark.shape
class TestEllipse(TestCase, BaseShapeTest):
    """Tests for the :code:`Ellipse` shape."""

    def setUp(self):
        self.shape = Ellipse(0, 0, 1 / 2, 1 / 3, 0.6)


@pytest.mark.shape
class TestCircle(TestCase, BaseShapeTest):
    """Tests for the :code:`Circle` shape."""

    def setUp(self):
        self.shape = Circle(0, 0, 0.5)


@pytest.mark.shape
class TestRectangle(TestCase, BaseShapeTest):
    """Tests for the :code:`Rectangle` shape."""

    def setUp(self):
        self.shape = Rectangle(0, 0, 0.6, 0.3)


@pytest.mark.shape
class TestSquare(TestCase, BaseShapeTest):
    """Tests for the :code:`Square` shape."""

    def setUp(self):
        self.shape = Square(0, 0, 0.75)

@pytest.mark.shape
class TestTriangle(TestCase, BaseShapeTest):
    """Tests for the :code:`Triangle` shape."""

    def setUp(self):
        self.shape = Triangle((1,.5),(.2,1),(.4,.5))
