def estimate_rsem_usage(gsm_dir, fastq2rsem_ratio): """ estimate the maximum disk space that is gonna be consumed by rsem analysis on one GSM based on a list of fq_gzs :param fq_gz_size: a number reprsenting the total size of fastq.gz files for the corresponding GSM """ fastq_usage = PPR.estimate_sra2fastq_usage(gsm_dir) return fastq_usage * fastq2rsem_ratio
def test_estimate_sra2fastq_usage(self, mock_ratio, mock_get_sras_info): mock_ratio = 2 mock_get_sras_info.return_value = PARSED_SRA_INFO_YAML_SINGLE_SRA self.assertEqual(ppr.estimate_sra2fastq_usage('some_gsm_dir'), 2546696608 * mock_ratio)