def setUp(self):
self.df1 = pd.DataFrame({"colA": [10, 12, 14, 15], "colB": [10, 12, 14, 15]})
self.df2 = pd.DataFrame({"colA": [10, 12, 14, 15], "colB": [10, 13, 14, 15]})
self.transform_result = pd.Series([0, 0, 0, 0])
self.RelativeDifference = RelativeDifference()
pass
class Test(unittest.TestCase):
def setUp(self):
self.df1 = pd.DataFrame({"colA": [10, 12, 14, 15], "colB": [10, 12, 14, 15]})
self.df2 = pd.DataFrame({"colA": [10, 12, 14, 15], "colB": [10, 13, 14, 15]})
self.transform_result = pd.Series([0, 0, 0, 0])
self.RelativeDifference = RelativeDifference()
pass
def tearDown(self):
pass
def test_RelativeDifference(self):
# do I really want to be pulling the column from the dataframe as a series
# just for this comparison? not sure, but it does run a valid test
assert (self.RelativeDifference.transform(self.df1).iloc[:, 0] == self.transform_result).all()
assert not (self.RelativeDifference.transform(self.df2).iloc[:, 0] == self.transform_result).all()