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)