def getSNPS(self, name, weight_db_logic): dosageLoader = None if self.input_format == Formats.IMPUTE: dosageLoader = ThousandGenomesUtilities.IMPUTEDosageLoader(self.data_folder, name) #outdated code elif self.input_format == Formats.PrediXcan: dosageName = Utilities.dosageName(name) path = os.path.join(self.data_folder, dosageName) dosageLoader = PrediXcanFormatUtilities.PrediXcanFormatDosageLoader(path, weight_db_logic) else: logging.info("Invalid input format: %s", self.input_format) return snps, snps_by_rsid = dosageLoader.load() return snps, snps_by_rsid
def testDosageName(self): dosage_name = Utilities.dosageName("a") self.assertEqual(dosage_name, "a.dosage.gz")