Пример #1
0
    def test_table_intergenic_warnings(self):
        var1 = Variant('var1')
        var1.output_type_flag = 'intergenic'
        var1.warnings = ['Validation error']
        var2 = Variant('var2')
        var2.output_type_flag = 'intergenic'
        var2.warnings = ['obsolete']
        var3 = Variant('var3')
        var3.output_type_flag = 'intergenic'
        var3.warnings = ['obsolete']
        var3.hgvs_transcript_variant = ''

        obj = ValOutput([var1, var2, var3], self.vv)
        res = obj.format_as_table(with_meta=False)
        self.assertIsInstance(res, list)
        self.assertEqual(res[1], [
            'var1', 'Validation error', None, None, None, None, None, None, '',
            '', '', '', '', '', '', '', '', '', '', '', '', '', '', '', ''
        ])
        self.assertEqual(res[2], [
            'var2', 'obsolete', None, None, None, None, None, None, '', '', '',
            '', '', '', '', '', '', '', '', '', '', '', '', '', ''
        ])
        self.assertEqual(res[3], [
            'var3', 'obsolete', '', None, None, None, None, None, '', '', '',
            '', '', '', '', '', '', '', '', '', '', '', '', '', ''
        ])
        self.assertEqual(len(res), 4)
Пример #2
0
    def test_dict_each_obsolete(self):
        var1 = Variant('var1')
        var1.warnings = ['obsolete']
        var1.output_type_flag = 'gene'
        var2 = Variant('var2')
        var2.warnings = ['obsolete']
        var2.output_type_flag = 'intergenic'
        var3 = Variant('var3')
        var3.warnings = ['obsolete']
        var4 = Variant('var4')
        var4.warnings = ['obsolete']
        var4.output_type_flag = 'gene'
        var4.hgvs_transcript_variant = ''

        obj = ValOutput([var1, var2, var3, var4], self.vv)
        res = obj.format_as_dict(with_meta=False)
        print(res)

        self.assertIsInstance(res, dict)
        self.assertEqual(list(res), [
            'flag', 'None', 'intergenic_variant_1', 'obsolete_record_1',
            'obsolete_record_2'
        ])
        self.assertEqual(res['flag'], 'gene_variant')
        self.assertEqual(res['None']['submitted_variant'], 'var1')
        self.assertEqual(res['intergenic_variant_1']['submitted_variant'],
                         'var2')
        self.assertEqual(res['obsolete_record_1']['submitted_variant'], 'var3')
        self.assertEqual(res['obsolete_record_2']['submitted_variant'], 'var4')