Adding test changes related to StudyMonth attribute

Author: anbhimi

tests/unit/test_cold_extraction.py

@@ -1,6 +1,8 @@
 import os
 import sys
+from charset_normalizer import logging
 import pytest
+import pdb
 
 from io import StringIO
 from pathlib import Path
@@ -73,21 +75,24 @@ def setup_method(self):
         Setup
         """
         CDR.csv_file = None
+        CDR.mod_csv_file = None
         CDR.extracted_ones = None
 
     def teardown_method(self):
         """
         Cleanup
         """
         CDR.csv_file = None
+        CDR.mod_csv_file = None
         CDR.extracted_ones = None
 
     def test_success(self):
         """
         Test for ColdDataRetriever.update_pickle function.
         Check whether function executes without raising error.
         """
-        CDR.csv_file = str(test_config.input_dir / 'csv_files' / 'tmp.csv')
+        # CDR.csv_file = str(test_config.input_dir / 'csv_files' / 'tmp.csv')
+        CDR.mod_csv_file = str(test_config.input_dir / 'csv_files' / 'tmp_mod.csv')
         CDR.extracted_ones = ['1234,4432']
         CDR.update_pickle()
         CDR.logging.debug.assert_called_once()
@@ -212,7 +217,7 @@ def setup_method(self):
         """
         Setup
         """
-        CDR.csv_file = str(test_config.input_dir /
+        CDR.mod_csv_file = str(test_config.input_dir /
                            'csv_files' / 'unit_test_read_csv.csv')
         CDR.number_of_query_attributes = 3
         self.reset_cdr_module_state()
@@ -244,7 +249,7 @@ def test_first_col_date(self):
         Checks read success by asserting value of number of rows read.
         """
         self.reset_cdr_module_state()
-        CDR.csv_file = str(test_config.input_dir /
+        CDR.mod_csv_file = str(test_config.input_dir /
                            'csv_files' / 'unit_test_read_csv_1st_date.csv')
         CDR.first_attr = "StudyDate"
         CDR.second_attr = "PatientID"
@@ -270,7 +275,7 @@ def test_third_col_date(self):
         Checks read success by asserting value of number of rows read.
         """
         self.reset_cdr_module_state()
-        CDR.csv_file = str(test_config.input_dir /
+        CDR.mod_csv_file = str(test_config.input_dir /
                            'csv_files' / 'unit_test_read_csv_3rd_date.csv')
 
         CDR.first_attr = "PatientID"