import pandas as pd

class CATEConversionEvaluateSimulationReady:

    def __init__(self,data_path):
        self.data_path = data_path

    def evaluate(self, discount_group):
        if discount_group == 5:
            qini_05_conversion_test = pd.read_csv(self.data_path + 'qini_05_conversion_test.csv').drop(columns='Unnamed: 0')
            qini_05_benefit_test = pd.read_csv(self.data_path + 'qini_05_benefit_test.csv').drop(columns='Unnamed: 0')
            return qini_05_conversion_test, qini_05_benefit_test
        elif discount_group == 10:
            qini_10_conversion_test = pd.read_csv(self.data_path + 'qini_10_conversion_test.csv').drop(columns='Unnamed: 0')
            qini_10_benefit_test = pd.read_csv(self.data_path + 'qini_10_benefit_test.csv').drop(columns='Unnamed: 0')
            return qini_10_conversion_test, qini_10_benefit_test
        elif discount_group == 15:
            qini_15_conversion_test = pd.read_csv(self.data_path + 'qini_15_conversion_test.csv').drop(columns='Unnamed: 0')
            qini_15_benefit_test = pd.read_csv(self.data_path + 'qini_15_benefit_test.csv').drop(columns='Unnamed: 0')
            return qini_15_conversion_test, qini_15_benefit_test
        
class CATEBenefitEvaluateSimulationReady:

    def __init__(self,data_path):
        self.data_path = data_path

    def evaluate(self, discount_group):
        if discount_group == 5:
            qini_05_conversion_test = pd.read_csv(self.data_path + 'qini_05_conversion_test_bate.csv').drop(columns='Unnamed: 0')
            qini_05_benefit_test = pd.read_csv(self.data_path + 'qini_05_benefit_test_bate.csv').drop(columns='Unnamed: 0')
            return qini_05_conversion_test, qini_05_benefit_test
        elif discount_group == 10:
            qini_10_conversion_test = pd.read_csv(self.data_path + 'qini_10_conversion_test_bate.csv').drop(columns='Unnamed: 0')
            qini_10_benefit_test = pd.read_csv(self.data_path + 'qini_10_benefit_test_bate.csv').drop(columns='Unnamed: 0')
            return qini_10_conversion_test, qini_10_benefit_test
        elif discount_group == 15:
            qini_15_conversion_test = pd.read_csv(self.data_path + 'qini_15_conversion_test_bate.csv').drop(columns='Unnamed: 0')
            qini_15_benefit_test = pd.read_csv(self.data_path + 'qini_15_benefit_test_bate.csv').drop(columns='Unnamed: 0')
            return qini_15_conversion_test, qini_15_benefit_test