diff options
| -rw-r--r-- | tests/integration/test_wgcna.py | 60 |
1 files changed, 48 insertions, 12 deletions
diff --git a/tests/integration/test_wgcna.py b/tests/integration/test_wgcna.py index 39dabb2..078449d 100644 --- a/tests/integration/test_wgcna.py +++ b/tests/integration/test_wgcna.py @@ -13,25 +13,61 @@ class WgcnaIntegrationTest(TestCase): self.app = create_app().test_client() @mock.patch("gn3.api.wgcna.call_wgcna_script") - def test_wgcna_endpoint(self, mock_wgcna_api): + def test_wgcna_endpoint(self, mock_wgcna_script): """test /api/wgcna/run_wgcna endpoint""" - wgcna_api_data = { - "eigengenes": ["1224_at", "121412_at", "32342342-at"], - "dendrogram_file_location": "/tmp/dend1.png" - + wgcna_output_data = { + "code": 0, + "output": "run script successfully", + "data": { + "ModEigens": { + "MEturquoise": [ + 0.0646677768085351, + 0.137200224277058, + 0.63451113720732, + -0.544002665501479, + -0.489487590361863, + 0.197111117570427 + ] + }, + "net_colors": { + "X1": "turquoise", + "X2": "turquoise", + "X3": "turquoise", + "X4": "turquoise" + }, + "imageLoc": "/WGCNAoutput_1uujpTIpC.png" + } } - mock_wgcna_api.return_value = wgcna_api_data request_data = { - - "trait_sample_data": [], - - + "trait_names": [ + "1455537_at", + "1425637_at" + ], + "trait_sample_data": [ + { + "129S1/SvImJ": 6.142, + "A/J": 5.31, + "AKR/J": 3.49, + "B6D2F1": 2.899, + "BALB/cByJ": 1.172, + "BALB/cJ": 7.396 + }, + { + "129S1/SvImJ": 1.42, + "A/J": 2.31, + "AKR/J": 5.49, + "B6D2F1": 3.899, + "BALB/cByJ": 1.172, + "BALB/cJ": 7.396 + } + ] } + mock_wgcna_script.return_value = wgcna_output_data response = self.app.post("/api/wgcna/run_wgcna", json=request_data, follow_redirects=True) - self.assertEqual(response.status_code, 401) - self.assertEqual(response.get_json(), wgcna_api_data) + self.assertEqual(response.status_code, 200) + self.assertEqual(response.get_json(), wgcna_output_data) |