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()