increase test coverage

VoSeq · Feb 3, 2015 · 3e03b97 · 3e03b97
1 parent 1cfa12b
commit 3e03b97
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/voseq/create_dataset/tests/tests_utils.py b/voseq/create_dataset/tests/tests_utils.py
@@ -15,7 +15,7 @@ def setUp(self):
 
         g1 = Genes.objects.get(gene_code='COI')
         g2 = Genes.objects.get(gene_code='EF1a')
-        cleaned_data = {
+        self.cleaned_data = {
             'gene_codes': [g1, g2],
             'taxonset': None,
             'voucher_codes': 'CP100-10\r\nCP100-11',
@@ -24,7 +24,7 @@ def setUp(self):
         }
 
         self.c = Client()
-        self.dataset_creator = CreateDataset(cleaned_data)
+        self.dataset_creator = CreateDataset(self.cleaned_data)
 
     def test_create_dataset(self):
         expected = '>CP100-10_Papilionoidea_Melitaea_diamina'
@@ -39,3 +39,14 @@ def test_get_taxon_names_for_taxa(self):
         result = self.dataset_creator.get_taxon_names_for_taxa()
 
         self.assertEqual(expected, result)
+
+    def test_get_taxon_names_for_taxa_additional_fields(self):
+        self.cleaned_data['taxon_names'] = ['SUPERFAMILY']
+        dataset_creator = CreateDataset(self.cleaned_data)
+        expected = {
+            'cp100-10': {'superfamily': 'Papilionoidea'},
+            'cp100-11': {'superfamily': ''},
+        }
+        result = dataset_creator.get_taxon_names_for_taxa()
+
+        self.assertEqual(expected, result)