예제 #1
0
    def test_create_table_if_different(self):
        so1 = StoredObject.create_table(self.wfm1, mock_csv_table)

        so2 = StoredObject.create_table_if_different(self.wfm1, so1,
                                                     mock_csv_table)
        self.assertIsNone(so2)

        so3 = StoredObject.create_table_if_different(self.wfm1, so1,
                                                     mock_csv_table2)
        self.assertIsNotNone(so3)
        table3 = so3.get_table()
        self.assertTrue(table3.equals(mock_csv_table2))
예제 #2
0
    def test_create_table_if_different(self):
        df1 = pd.DataFrame({"A": [1]})
        df2 = pd.DataFrame({"A": [2]})

        so1 = StoredObject.create_table(self.wfm1, df1)

        so2 = StoredObject.create_table_if_different(self.wfm1, so1, df1)
        self.assertIsNone(so2)

        so3 = StoredObject.create_table_if_different(self.wfm1, so1, df2)
        self.assertIsNotNone(so3)

        table3 = so3.get_table()
        assert_frame_equal(table3, df2)