aboutsummaryrefslogtreecommitdiff
path: root/tests/integration/test_wgcna.py
diff options
context:
space:
mode:
authorAlexander Kabui2021-09-27 19:15:23 +0300
committerAlexander Kabui2021-09-27 19:15:23 +0300
commita2da1f5dbc49b0137ef6b8ee9e234178521935f3 (patch)
treecda90a1b31509258bd6637cc13fad151b4d7fc5d /tests/integration/test_wgcna.py
parentb6097cff6431e50c132f46dc0d3e1b841897da0f (diff)
downloadgenenetwork3-a2da1f5dbc49b0137ef6b8ee9e234178521935f3.tar.gz
modify integration tests
Diffstat (limited to 'tests/integration/test_wgcna.py')
-rw-r--r--tests/integration/test_wgcna.py60
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)