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)
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')