# -------------------------------------
# File: Xiang/002.py
# Author: Mihaela Malita
# Title: OO Shapes: Polymorphism: area of circle and square
# Run:
# area of shape is 1
# area of shape is 25
# area of shape is 3.14159265359
# area of shape is 78.5398163397
# -------------------------------------
from math import pi
class mySquare:
def __init__(self,s):
self.side = s
def area(self):
return self.side ** 2
class myCircle:
def __init__(self,r):
self.radius = r
def area(self):
return pi * self.radius ** 2
def main():
# make a list of shapes
mylist= [mySquare(1),mySquare(5),myCircle(1),myCircle(5)]
# print the coresponding areas
for i in mylist:
print "area of shape is ", i.area()
main()