From 805149654a349d5a94af35a081f160771f73219c Mon Sep 17 00:00:00 2001
From: Venkat Malladi <venkat.malladi@utsouthwestern.edu>
Date: Sat, 29 Feb 2020 09:54:04 -0600
Subject: [PATCH] Add in tests for checking single control se and pe.

---
 test_data/A_1.bedpe.gz                        | Bin 0 -> 102 bytes
 test_data/B_1.bedpe.gz                        | Bin 0 -> 102 bytes
 .../tests/test_pool_and_psuedoreplicate.py    |  19 ++++++++++++++++--
 3 files changed, 17 insertions(+), 2 deletions(-)
 create mode 100644 test_data/A_1.bedpe.gz
 create mode 100644 test_data/B_1.bedpe.gz

diff --git a/test_data/A_1.bedpe.gz b/test_data/A_1.bedpe.gz
new file mode 100644
index 0000000000000000000000000000000000000000..52beb16aa36e95cb5e14dfd4d7fb867f6b4183bc
GIT binary patch
literal 102
zcmV-s0Ga<EiwFpKhgx0$13_OgE@EY5aAg2X&L}ceP%tzwP*6}X0+Y!ZMWzZ0rUoF6
zDOe&YHKiamKD8n-w;(4KsM;8$Oj|)gmkX%Y60F+*rUN2k3}zc!VCXi2+6C4P08WjC
Ig?s=207DBPy8r+H

literal 0
HcmV?d00001

diff --git a/test_data/B_1.bedpe.gz b/test_data/B_1.bedpe.gz
new file mode 100644
index 0000000000000000000000000000000000000000..52beb16aa36e95cb5e14dfd4d7fb867f6b4183bc
GIT binary patch
literal 102
zcmV-s0Ga<EiwFpKhgx0$13_OgE@EY5aAg2X&L}ceP%tzwP*6}X0+Y!ZMWzZ0rUoF6
zDOe&YHKiamKD8n-w;(4KsM;8$Oj|)gmkX%Y60F+*rUN2k3}zc!VCXi2+6C4P08WjC
Ig?s=207DBPy8r+H

literal 0
HcmV?d00001

diff --git a/workflow/tests/test_pool_and_psuedoreplicate.py b/workflow/tests/test_pool_and_psuedoreplicate.py
index 7f219c9..b28c59c 100644
--- a/workflow/tests/test_pool_and_psuedoreplicate.py
+++ b/workflow/tests/test_pool_and_psuedoreplicate.py
@@ -33,8 +33,11 @@ def design_experiment_2(design_experiment):
 
 @pytest.fixture
 def design_experiment_3(design_experiment):
-    # Update second control to be same as first
-    design_experiment.loc[1, 'control_tag_align'] = 'B_1.bedse.gz'
+    # Drop Replicate A_2
+    design_df = design_experiment.drop(design_experiment.index[1])
+    # Update to be paired as first
+    design_experiment.loc[0, 'control_tag_align'] = 'B_1.bedpe.gz'
+	design_experiment.loc[0, 'ttag_align'] = 'A_1.bedpe.gz'
     return design_experiment
 
 
@@ -71,6 +74,18 @@ def test_single_rep(design_experiment_2):
     shutil.copy(test_design_path + 'B_1.tagAlign.gz', cwd)
     single_rep = pool_and_psuedoreplicate.generate_design('false', 1.2, design_experiment_2, cwd, 1, 1)
     assert single_rep.shape[0] == 4
+	assert len(single_rep['control_tag_align'].unique()) == 1
+	assert single_rep['control_tag_align'].unique()[0] == 'B_1.tagAlign.gz'
+
+
+@pytest.mark.unit
+def test_single_control(design_experiment_3):
+    cwd = os.getcwd()
+    shutil.copy(test_design_path + 'A_1.bedpe.gz', cwd)
+    shutil.copy(test_design_path + 'B_1.bedpe.gz', cwd)
+    shutil.copy(test_design_path + 'A_1.tagAlign.gz', cwd)
+    single_control = pool_and_psuedoreplicate.generate_design('true', 1.2, design_experiment_3, cwd, 1, 1)
+	assert single_control['control_tag_align'].unique() == 'B_1.tagAlign.gz'
 
 
 @pytest.mark.singleend
-- 
GitLab