def setUpClass(cls): #---------------------- # set up db #---------------------- cls.db = ks_db_settings.connect() cls.ks_fh = filehandler(cls.db) cls.ks_fh.reset() cls.company_id = 1 register_raw_filesCsvPy("Sales_new_version",cls.company_id, cls.db) register_raw_filesCsvPy("CurrencyV2",cls.company_id, cls.db) register_raw_filesCsvPy("ComissionTax_new_version",cls.company_id, cls.db) register_raw_filesCsvPy("CountryRegion",cls.company_id, cls.db) sql = 'update files set file_name = "./tests/data2/version/Sales.csv" where file_name = "./tests/data2/Sales.csv"' cls.db.cursor().execute(sql) sql = 'update files set file_name = "./tests/data2/version/ComissionTax.csv" where file_name = "./tests/data2/ComissionTax.csv"' cls.db.cursor().execute(sql) ks_precompute = precompute(cls.db) ks_precompute.reset() #precompute load_precompute_normalize_CsvPy(cls.company_id, cls.db) newBigTable = "BigTable"+ str(ks_precompute.getMaxBigTableIdForCompany(cls.company_id)) cls.ks_analytics = analytics(cls.db) #clean up sql ="update %s set TaxRate = TaxRate/100;"%("analytics."+newBigTable) cls.db.cursor().execute(sql) # ProductType changed from D to M see documentation of test case sql ="update %s set ProductType = 'M' where VendorId='0268_20140114_SOFA_ENGLIS' and DownloadDate='6/1/14';"%("analytics."+newBigTable) cls.db.cursor().execute(sql) cls.ks_analytics.addFactUsingBinaryOpAPI("NET_REVENUE", "Units", "RoyaltyPrice", "*", newBigTable) cls.ks_analytics.addFactUsingBinaryOpAPI("TAXES", "NET_REVENUE","TaxRate","*", newBigTable) cls.ks_analytics.addFactUsingBinaryOpAPI("REVENUE_AFTER_TAX", "NET_REVENUE","TAXES","-", newBigTable) cls.ks_fh.registerFormula("", "Plus", "Plus", "Units+RoyaltyPrice", "sum") cls.ks_fh.registerFormula("", "Mult", "Mult", "Units*RoyaltyPrice", "sum") cls.ks_fh.registerFormula("", "Individual_Tax", "Individual_Tax", "RoyaltyPrice*TaxRate", "sum") cls.ks_fh.registerFormula("", "NET_REVENUE", "NET_REVENUE", "Units*RoyaltyPrice", "sum") cls.ks_fh.registerFormula("", "SumPlus", "SumPlus", "Sum(Units)+Sum(RoyaltyPrice)", "sum") cls.ks_fh.registerFormula("", "SumMult", "SumMult", "Sum(Units)*Sum(RoyaltyPrice)", "sum") cls.ks_fh.registerFormula("", "Individual_TaxSum", "Individual_TaxSum", "Sum(RoyaltyPrice)*Sum(TaxRate)", "sum") cls.ks_fh.registerFormula("", "NonsenseSum", "Individual_Tax", "Sum(RoyaltyPrice)+Sum(TaxRate)", "sum") cls.ks_fh.registerFormula("", "Nonsense", "Individual_Tax", "RoyaltyPrice+TaxRate", "sum") cls.ks_fh.registerFormula("", "REVENUE_AFTER_TAX", "REVENUE_AFTER_TAX", "", "sum")
def setUpClass(cls): #---------------------- # set up db #---------------------- cls.db = ks_db_settings.connect() cls.ks_fh = filehandler(cls.db) cls.ks_fh.reset() cls.company_id = 1 register_raw_filesCsvPy("Sales",cls.company_id, cls.db) register_raw_filesCsvPy("CurrencyV2",cls.company_id, cls.db) register_raw_filesCsvPy("ComissionTax",cls.company_id, cls.db) register_raw_filesCsvPy("CountryRegion",cls.company_id, cls.db) ks_precompute = precompute(cls.db) ks_precompute.reset() #precompute load_precompute_normalize_CsvPy(cls.company_id, cls.db) newBigTable = "BigTable"+ str(ks_precompute.getMaxBigTableIdForCompany(cls.company_id)) cls.ks_analytics = analytics(cls.db) cls.ks_fh.registerFormula("", "UnitsSUM", "UnitsSUM", "sum(Units)", "sum")
def setUpClass(cls): #---------------------- # set up db #---------------------- cls.db = ks_db_settings.connect() cls.ks_fh = filehandler(cls.db) cls.ks_fh.reset() cls.company_id = 1 register_raw_filesCsvPy("Salesfreq",cls.company_id, cls.db) #register_raw_filesCsvPy("CurrencyV2",cls.company_id, cls.db) register_raw_filesCsvPy("ComissionTax",cls.company_id, cls.db) register_raw_filesCsvPy("CountryRegion",cls.company_id, cls.db) ks_precompute = precompute(cls.db) ks_precompute.reset() #precompute load_precompute_normalize_CsvPy(cls.company_id, cls.db) newBigTable = "BigTable"+ str(ks_precompute.getMaxBigTableIdForCompany(cls.company_id)) cls.ks_analytics = analytics(cls.db) cls.ks_analytics.addFactUsingBinaryOpAPI("NET_REVENUE", "Units", "RoyaltyPrice", "*", newBigTable) cls.ks_analytics.addFactUsingBinaryOpAPI("TAXES", "NET_REVENUE","TaxRate","*", newBigTable) cls.ks_analytics.addFactUsingBinaryOpAPI("REVENUE_AFTER_TAX", "NET_REVENUE","TAXES","-", newBigTable) cls.ks_fh.registerFormula("", "Plus", "Plus", "Units+RoyaltyPrice", "sum") cls.ks_fh.registerFormula("", "Mult", "Mult", "Units*RoyaltyPrice", "sum") cls.ks_fh.registerFormula("", "Individual_Tax", "Individual_Tax", "RoyaltyPrice*TaxRate", "sum") cls.ks_fh.registerFormula("", "NET_REVENUE", "NET_REVENUE", "Units*RoyaltyPrice", "sum") cls.ks_fh.registerFormula("", "SumPlus", "SumPlus", "Sum(Units)+Sum(RoyaltyPrice)", "sum") cls.ks_fh.registerFormula("", "SumMult", "SumMult", "Sum(Units)*Sum(RoyaltyPrice)", "sum") cls.ks_fh.registerFormula("", "Individual_TaxSum", "Individual_TaxSum", "Sum(RoyaltyPrice)*Sum(TaxRate)", "sum") cls.ks_fh.registerFormula("", "NonsenseSum", "Individual_Tax", "Sum(RoyaltyPrice)+Sum(TaxRate)", "sum") cls.ks_fh.registerFormula("", "Nonsense", "Individual_Tax", "RoyaltyPrice+TaxRate", "sum") cls.ks_fh.registerFormula("", "REVENUE_AFTER_TAX", "REVENUE_AFTER_TAX", "", "sum")
# import API from register_raw_files import register_raw_filesCsvPy from load_precompute_normalize import load_precompute_normalize_CsvPy def MeasureName2MeasureIds(name): measure_id = ks_fh.getMeasureID(name) measure_ids = [] measure_ids.append(measure_id) return measure_ids #---------------------- # set up db #---------------------- db = ks_db_settings.connect() ks_fh = filehandler(db) ks_fh.reset() company_id = 1 register_raw_filesCsvPy("Sales",company_id, db) register_raw_filesCsvPy("CurrencyV2",company_id, db) register_raw_filesCsvPy("ComissionTax",company_id, db) register_raw_filesCsvPy("CountryRegion",company_id, db) ks_precompute = precompute(db) ks_precompute.reset() #precompute load_precompute_normalize_CsvPy(company_id, db) newBigTable = "BigTable"+ str(ks_precompute.getMaxBigTableIdForCompany(company_id))