Python Programming - Polymorphism - Discussion

class Shape:
    def draw(self):
        return "Drawing a shape"

class Circle(Shape):
    def draw(self):
        return "Drawing a circle"

class Square(Shape):
    def draw(self):
        return "Drawing a square"

def display_shape_info(shape):
    return shape.draw()

circle = Circle()
square = Square()

print(display_shape_info(circle))
print(display_shape_info(square))