This worksheet focuses on object-oriented programming.
(While last week included some lecture material on higher-order functions and lamdba, they are not included on this worksheet because you'll work with those topics a lot on Project 3. But you can check out the higher/ folder in the course sample code repo to see several examples, including ones that were mentioned in lecture but not covered in detail.)
Following the new policy, Problem 1 is different in that:
The main course materials to refer to for this worksheet are:
(Lecture videos are not linked on worksheets, but are also useful to review while working on worksheets. Video links can be found in the course course Blackboard site.)
Analogous to the classes discussed in our OOP lectures (for geometric objects Circle and Rectangle), create a Python class called Segment to store a line segment in the plane. The constructor should accept four arguments: x0,y0,x1,y1. The point (x0,y0) is then one endpoint of the segment, and (x1,y1) is the other endpoint. All four constructor arguments should be stored as attributes.
The following methods should be included:
translate(dx,dy)
- move the segment horizontally by dx
and vertically by dy
. This should modify the object, not returning anything.scale(factor)
- increase the length of the segment by a factor of factor
, keeping the center point the same. This is tricky, because the center point is not part of the data stored in the class! Feel free to skip this at first and come back to it later.__str__()
- make a reasonable human-readable string representation that includes the values of x0,y0,x1,y1.length()
- returns a float, the distance between the endpoints of the segment.Also, read the next problem before you start work on this.
import math # so that sqrt is available
class Segment():
'''Creates a 2D line segment by storing the endpoints as attributes'''
def __init__(self, x0, y0, x1, y1):
'''Constructor stores the endpoint values to the Segment object'''
self.x0 = x0
self.y0 = y0
self.x1 = x1
self.y1 = y1
def translate(self, dx, dy):
'''Moves the Segment horizontally by dx, and vertically by dy. Returns nothing'''
self.x0 += dx
self.y0 += dy
self.x1 += dx
self.y1 += dy
def scale(self, factor):
'''Scales the Segment by `factor` while keeping the center point the same. Returns nothing'''
# First, calculate the center of the segment using the averages of the endpoints
centerx = (self.x0 + self.x1)/2
centery = (self.y0 + self.y1)/2
# Translate the segment so it is centered at the origin (0,0)
self.translate(-centerx, -centery)
# Scale the segment by multiplying the endpoint values by factor
self.x0 *= factor
self.x1 *= factor
self.y0 *= factor
self.y1 *= factor
# Translate the scaled segment back to its original center
self.translate(centerx, centery)
def __str__(self):
'''Human-readable string representing the Segment'''
return "Segment(x0={},y0={},x1={},y1={})".format(self.x0, self.y0, self.x1, self.y1)
def __repr__(self):
'''Unambiguous string representation shown in REPL'''
return self.__str__()
def length(self):
'''Returns the length of the segment calculated with the Pythagorean Thm'''
dx = self.x1 - self.x0
dy = self.y1 - self.y0
return math.sqrt(dx**2 + dy**2)
This problem builds on the Segment class from problem 1.
Suppose the Segment class from Problem 2 is stored in a file geom.py
. Write a program to test the Segment class. It should do the following:
Any time you are tempted to test whether two floats are equal, please instead use a check of whether they differ by a very small amount, e.g. instead of
if x == y:
# stuff
use
if abs(x-y) < 0.000000001:
# stuff
This will help to avoid problems created by the fact that float operations are only approximations of the associated operations on real numbers. For example,
0.1 + 0.2 == 0.3
evaluates to False
because of the slight error in float addition (compared to real number addition), whereas
abs( 0.3 - (0.1+0.2) ) < 0.000000001
evaluates to True
.
import geom
print("Create a Segment L from (0,0) to (0,4)")
L = geom.Segment(0, 0, 0, 4)
print("Confirm that L's endpoints are accessible")
print(L)
print(L.x0, L.y0, L.x1, L.y1)
print("Confirm that length is 4: ", abs(L.length()-4)<1e-12)
print("Try to scale L by 2 to (0,0),(0,8)")
L.scale(2)
print(L)
print("Confirm that length is 8: ", abs(L.length()-8)<1e-12)
print("Try to translate L to (50,100),(50,108)")
L.translate(50,100)
print(L)
Below you will find code that defines a class QuantityWithUnit
that is meant to store float quantities that also have a unit of measurement attached to them, such as 55 m
(55 meters), 2.8 s
(2.8 seconds), or 94.05 kg
(94.05 kilograms). Save it in a file qwu.py
on your computer, so you can import it with import qwu
. Read the code and familiarize yourself with what it does. Then try the following:
The final product of your work on this question could be a program that demonstrates all of these features.
class QuantityWithUnit:
"""A float quantity that also has a unit name, such as
kilograms, meters, seconds, etc."""
def __init__(self,qty,unit):
"""Create new quantity with units"""
self.qty = float(qty)
self.unit = unit
def __str__(self):
"""Make human-readable string version of quantity"""
return "{} {}".format(self.qty,self.unit)
def __repr__(self):
"""Make human-readable string version of quantity"""
return "QuantityWithUnit(qty={},unit={})".format(self.qty,self.unit)
def __add__(self,other):
"""Sum of two quantities requires them to have the same units"""
if not isinstance(other,QuantityWithUnit):
raise TypeError("Can only add a QuantityWithUnit to another QuantityWithUnit")
if self.unit != other.unit:
raise ValueError("Cannot add quantities with different units: {} and {}".format(self,other))
return QuantityWithUnit(self.qty+other.qty,self.unit)
def __sub__(self,other):
"""Difference of two quantities requires them to have the same units"""
if not isinstance(other,QuantityWithUnit):
raise TypeError("Can only subtract a QuantityWithUnit from another QuantityWithUnit")
if self.unit != other.unit:
raise ValueError("Cannot subtract quantities with different units: {} and {}".format(self,other))
return QuantityWithUnit(self.qty-other.qty,self.unit)
import qwu
M = qwu.QuantityWithUnit(19, "kg")
print("M",M)
print("M.__str__()",M) # How print prints M
print("M.__repr__()",M.__repr__()) # How the REPL would show M
print("M+M",M+M)
print("M-M",M-M)
T = qwu.QuantityWithUnit(3600, "sec")
print("M+T",M+T)
You should look at the previous problem before attempting this one, because it presumes familiarity with the QuantityWithUnit
class.
First, add support for testing equality to QuantityWithUnit
. Two quantities with unit should be considered equal if the float quantities are equal, and if the units are equal.
Now, consider adding support for multiplication as follows.
Multiplying two quantities with units results in an answer that has a different unit. For example, 11 kilograms multiplied by 20 seconds is equal to 220 kilogram-seconds.
However, it does make sense to multiply a quantity with units by a number that has no units at all. For example, if you have 16 apples that each have a mass of 0.1 kilograms, then the total mass is (0.1kg)*16 = 1.6kg.
Add an operator overloading feature to QuantityWithUnit
that allows such a quantity to be multiplied by a number as long as it is not an instance of QuantityWithUnit
. If you do this correctly, then the following tests should behave as the comments suggest. These assume that you have QuantityWithUnit
in a module called qwu
.
# content of qwu.py
# slightly modified from the class given in problem 3
class QuantityWithUnit:
"""A float quantity that also has a unit name, such as
kilograms, meters, seconds, etc."""
def __init__(self,qty,unit):
"""Create new quantity with units"""
self.qty = float(qty)
self.unit = unit
def __str__(self):
"""Make human-readable string version of quantity"""
return "{} {}".format(self.qty,self.unit)
def __repr__(self):
"""Make human-readable string version of quantity"""
return "QuantityWithUnit(qty={},unit={})".format(self.qty,self.unit)
def __add__(self,other):
"""Sum of two quantities requires them to have the same units"""
if not isinstance(other,QuantityWithUnit):
raise TypeError("Can only add a QuantityWithUnit to another QuantityWithUnit")
if self.unit != other.unit:
raise ValueError("Cannot add quantities with different units: {} and {}".format(self,other))
return QuantityWithUnit(self.qty+other.qty,self.unit)
def __sub__(self,other):
"""Difference of two quantities requires them to have the same units"""
if not isinstance(other,QuantityWithUnit):
raise TypeError("Can only subtract a QuantityWithUnit from another QuantityWithUnit")
if self.unit != other.unit:
raise ValueError("Cannot subtract quantities with different units: {} and {}".format(self,other))
return QuantityWithUnit(self.qty-other.qty,self.unit)
# ----------------------
# NEW STUFF STARTS HERE
# ----------------------
def __eq__(self, other):
"""Returns a True if the other is a QWU with the same qty and unit, False otherwise"""
return self.__repr__() == other.__repr__()
def __mul__(self, factor):
"""Returns a new QWU with the same units and the qty scaled by `factor` which must be a scalar."""
try:
# If `factor` is something we can convert to a float, we consider it
# to be an acceptable thing to multiply by
factor = float(factor)
except TypeError:
raise TypeError("Can only multiply a QuantityWithUnit by a scalar factor")
return QuantityWithUnit(self.qty*factor, self.unit)
import qwu
car_mass = qwu.QuantityWithUnit(1200,"kg")
car_mass_2 = qwu.QuantityWithUnit(1200,"kg")
person_mass = qwu.QuantityWithUnit(68,"kg")
lecture_length = qwu.QuantityWithUnit(50,"min")
print("car_pass+person_mass:",car_mass + person_mass) # works, have same units
print("car_pass-person_mass:",car_mass - person_mass) # works, have same units
print("car_mass==person_mass:",car_mass == person_mass) # works, should print False
print("car_mass==car_mass_2:",car_mass == car_mass_2) # works, should print True
print("car_mass*5:",car_mass * 5) # works -- mass of 5 cars
print("car_mass*lecture_length:",car_mass * lecture_length) # fails (exception raised); only allowed to multiply by unitless numbers