aboutsummaryrefslogtreecommitdiff
path: root/wqflask
diff options
context:
space:
mode:
Diffstat (limited to 'wqflask')
-rw-r--r--wqflask/tests/wqflask/show_trait/test_show_trait.py47
1 files changed, 20 insertions, 27 deletions
diff --git a/wqflask/tests/wqflask/show_trait/test_show_trait.py b/wqflask/tests/wqflask/show_trait/test_show_trait.py
index 1b5d2265..408797b9 100644
--- a/wqflask/tests/wqflask/show_trait/test_show_trait.py
+++ b/wqflask/tests/wqflask/show_trait/test_show_trait.py
@@ -20,6 +20,7 @@ from wqflask.show_trait.show_trait import get_nearest_marker
from wqflask.show_trait.show_trait import get_genotype_scales
+
class TraitObject:
def __init__(self, obj):
for key, value in obj.items():
@@ -141,7 +142,7 @@ class TestTraits(unittest.TestCase):
def test_get_table_widths(self):
"""test for getting table widths"""
sample_groups = [TraitObject({'se_exists': True, "attributes": ["attr1", "attr2", "attr3"]}
- ), TraitObject(
+ ), TraitObject(
{"se_exists": False, "attributes": ["at1", "at2"]
})]
@@ -221,47 +222,39 @@ class TestTraits(unittest.TestCase):
mock_db.db.execute.assert_called_once()
self.assertEqual(results_empty_db, "")
-
@mock.patch("wqflask.show_trait.show_trait.get_scales_from_genofile")
- def test_get_genotype_scales_with_genofile_is_list(self,mock_get_scales):
+ def test_get_genotype_scales_with_genofile_is_list(self, mock_get_scales):
"""test for getting genotype scales with genofile as list """
- #where genofile is instance of list
- genofiles_list = [{"filename":"file1","location":"~/data/files/f1"},{"filename":"file2","location":"~/data/files/f2"},{"filename":"file3","location":"~/data/files/f3"}]
+ # where genofile is instance of list
+ genofiles_list = [{"filename": "file1", "location": "~/data/files/f1"},
+ {"filename": "file2", "location": "~/data/files/f2"},
+ {"filename": "file3", "location": "~/data/files/f3"}]
- mock_get_scales.side_effect = [[["morgan", "cM"]],[["morgan", "cM"]],[["physic", "Mb"]]]
+ mock_get_scales.side_effect = [[["morgan", "cM"]],
+ [["morgan", "cM"]],
+ [["physic", "Mb"]]]
results = get_genotype_scales(genofiles_list)
expected_results = {
- "~/data/files/f1":[["morgan","cM"]],
- "~/data/files/f2":[["morgan","cM"]],
- "~/data/files/f3":[["physic","Mb"]]
+ "~/data/files/f1": [["morgan", "cM"]],
+ "~/data/files/f2": [["morgan", "cM"]],
+ "~/data/files/f3": [["physic", "Mb"]]
}
- multiple_calls = [mock.call('~/data/files/f1'),mock.call('~/data/files/f2'),
- mock.call('~/data/files/f3')]
-
+ multiple_calls = [mock.call('~/data/files/f1'), mock.call('~/data/files/f2'),
+ mock.call('~/data/files/f3')]
mock_get_scales.assert_has_calls(multiple_calls)
- self.assertEqual(results,expected_results)
-
+ self.assertEqual(results, expected_results)
@mock.patch("wqflask.show_trait.show_trait.get_scales_from_genofile")
- def test_genotype_scales_with_genofile_other(self,mock_get_scales):
+ def test_genotype_scales_with_genofile_other(self, mock_get_scales):
"""test for getting genotype scales with genofile as a string"""
file_location = "~/another_file_location"
- mock_get_scales.return_value = [["physic","Mb"]]
+ mock_get_scales.return_value = [["physic", "Mb"]]
- expected_results = {f"{file_location}":[["physic","Mb"]]}
+ expected_results = {f"{file_location}": [["physic", "Mb"]]}
-
- self.assertEqual(get_genotype_scales(file_location),expected_results)
+ self.assertEqual(get_genotype_scales(file_location), expected_results)
mock_get_scales.assert_called_once_with(file_location)
-
-
-
-
-
-
-
-