aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/test_file_utils.py
diff options
context:
space:
mode:
authorBonfaceKilz2021-02-12 15:35:43 +0300
committerBonfaceKilz2021-02-12 15:50:37 +0300
commitec5cb3e65df225cd589ea63effb0a26b4201c428 (patch)
tree6e44c8399270a57332b50dcb3fbd9039e85fddbe /tests/unit/test_file_utils.py
parent21a4a847456fde5fcc6072df0d0fc36992da283d (diff)
downloadgenenetwork3-ec5cb3e65df225cd589ea63effb0a26b4201c428.tar.gz
Add new procedure for looking up files in a given ENV variable
* gn3/file_utils.py (lookup_file): New function. * tests/unit/test_file_utils.py: New test cases for ^^.
Diffstat (limited to 'tests/unit/test_file_utils.py')
-rw-r--r--tests/unit/test_file_utils.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/tests/unit/test_file_utils.py b/tests/unit/test_file_utils.py
index e6109bb..231dcd0 100644
--- a/tests/unit/test_file_utils.py
+++ b/tests/unit/test_file_utils.py
@@ -2,6 +2,8 @@
import os
import unittest
+from unittest import mock
+from gn3.file_utils import lookup_file
from gn3.file_utils import get_dir_hash
@@ -18,3 +20,31 @@ class TestFileUtils(unittest.TestCase):
self.assertRaises(FileNotFoundError,
get_dir_hash,
"/non-existent-file")
+
+ @mock.patch("os.path.isfile")
+ @mock.patch.dict(os.environ, {"GENENETWORK_FILES": "/tmp/"})
+ def test_lookup_genotype_file_exists(self, mock_isfile):
+ """Test whether genotype file exists if file is present"""
+ mock_isfile.return_value = True
+ self.assertEqual(lookup_file("GENENETWORK_FILES",
+ "genotype_files", "genotype.txt"),
+ "/tmp/genotype_files/genotype.txt")
+
+ @mock.patch("os.path.isfile")
+ @mock.patch.dict(os.environ, {"GENENETWORK_FILES": "/tmp"})
+ def test_lookup_genotype_file_does_not_exist(self, mock_isfile):
+ """Test whether genotype file exists if file is absent"""
+ mock_isfile.return_value = False
+ self.assertRaises(FileNotFoundError,
+ lookup_file,
+ "GENENETWORK_FILES",
+ "genotype_files",
+ "genotype.txt")
+
+ def test_lookup_genotype_file_env_does_not_exist(self):
+ """Test whether genotype file exists if GENENETWORK_FILES is absent"""
+ self.assertRaises(FileNotFoundError,
+ lookup_file,
+ "GENENETWORK_FILES",
+ "genotype_files",
+ "genotype.txt")