From 302a2915be6ff395e60f6d5e6841cc4523e99185 Mon Sep 17 00:00:00 2001
From: Venkat Malladi <venkat.malladi@utsouthwestern.edu>
Date: Fri, 19 Apr 2019 12:33:02 -0500
Subject: [PATCH] Update tests and import of yaml.

---
 workflow/tests/test_annotate_peaks.py               | 2 +-
 workflow/tests/test_generate_software_references.py | 1 +
 workflow/tests/test_generate_software_versions.py   | 1 +
 workflow/tests/test_overlap_peaks.py                | 2 +-
 4 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/workflow/tests/test_annotate_peaks.py b/workflow/tests/test_annotate_peaks.py
index 1342a12..3df797b 100644
--- a/workflow/tests/test_annotate_peaks.py
+++ b/workflow/tests/test_annotate_peaks.py
@@ -24,7 +24,7 @@ def test_upsetplot_singleend():
 def test_annotation_singleend():
     annotation_file = test_output_path + 'ENCSR238SGC.chipseeker_annotation.tsv'
     assert os.path.exists(annotation_file)
-    assert utils.count_lines(annotation_file) == 152255
+    assert utils.count_lines(annotation_file) == 149284
 
 
 @pytest.mark.pairedend
diff --git a/workflow/tests/test_generate_software_references.py b/workflow/tests/test_generate_software_references.py
index 3b6fe20..a627887 100644
--- a/workflow/tests/test_generate_software_references.py
+++ b/workflow/tests/test_generate_software_references.py
@@ -3,6 +3,7 @@
 import pytest
 import os
 import utils
+import yaml
 
 test_output_path = os.path.dirname(os.path.abspath(__file__)) + \
                 '/../output/softwareReport/'
diff --git a/workflow/tests/test_generate_software_versions.py b/workflow/tests/test_generate_software_versions.py
index 77489a8..276fc96 100644
--- a/workflow/tests/test_generate_software_versions.py
+++ b/workflow/tests/test_generate_software_versions.py
@@ -3,6 +3,7 @@
 import pytest
 import os
 import utils
+import yaml
 
 test_output_path = os.path.dirname(os.path.abspath(__file__)) + \
                 '/../output/softwareReport/'
diff --git a/workflow/tests/test_overlap_peaks.py b/workflow/tests/test_overlap_peaks.py
index ff61c93..411bfd0 100644
--- a/workflow/tests/test_overlap_peaks.py
+++ b/workflow/tests/test_overlap_peaks.py
@@ -37,7 +37,7 @@ def test_check_update_design(design_diff):
 def test_overlap_peaks_singleend():
     assert os.path.exists(os.path.join(test_output_path, 'ENCSR238SGC.rejected.narrowPeak'))
     peak_file = test_output_path + 'ENCSR238SGC.replicated.narrowPeak'
-    assert utils.count_lines(peak_file) == 152262
+    assert utils.count_lines(peak_file) == 149291
 
 
 @pytest.mark.pairedend
-- 
GitLab