Making import pytest to global

Author: rahulr-NI

generated/nifake/nifake/unit_tests/test_session.py

@@ -4,6 +4,7 @@
 import nifake
 import nifake.errors
 import numpy
+import pytest
 
 from unittest.mock import MagicMock
 from unittest.mock import patch
@@ -847,7 +848,6 @@ def test_function_with_int_flag_parameter_with_specific_flags(self):
 
     def test_function_with_int_flag_parameter_invalid_type(self):
         invalid_flag = "not_an_intflag"
-        import pytest
         with nifake.Session('dev1') as session:
             with pytest.raises(TypeError) as exc_info:
                 session.function_with_int_flag_parameter(invalid_flag)
@@ -857,7 +857,6 @@ def test_function_with_int_flag_parameter_invalid_type(self):
     def test_session_write_waveform_numpy_complex64_invalid_dtype(self):
         invalid_waveform_data = numpy.full(10, 1.0 + 1.0j, dtype=numpy.complex128)
         expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=complex64, is complex128"
-        import pytest
         with nifake.Session('dev1') as session:
             with pytest.raises(TypeError) as exc_info:
                 session.write_waveform_numpy_complex64(invalid_waveform_data)
@@ -866,7 +865,6 @@ def test_session_write_waveform_numpy_complex64_invalid_dtype(self):
     def test_session_write_waveform_numpy_complex128_invalid_dtype(self):
         invalid_waveform_data = numpy.full(10, 1.0 + 1.0j, dtype=numpy.complex64)
         expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=complex128, is complex64"
-        import pytest
         with nifake.Session('dev1') as session:
             with pytest.raises(TypeError) as exc_info:
                 session.write_waveform_numpy_complex128(invalid_waveform_data)
@@ -875,7 +873,6 @@ def test_session_write_waveform_numpy_complex128_invalid_dtype(self):
     def test_session_write_waveform_numpy_complex_interleaved_i16_invalid_dtype(self):
         invalid_waveform_data = numpy.full(10, 1.0 + 1.0j, dtype=numpy.complex64)
         expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=int16, is complex64"
-        import pytest
         with nifake.Session('dev1') as session:
             with pytest.raises(TypeError) as exc_info:
                 session.write_waveform_numpy_complex_interleaved_i16(invalid_waveform_data)

src/nifake/unit_tests/test_session.py

@@ -4,6 +4,7 @@
 import nifake
 import nifake.errors
 import numpy
+import pytest
 
 from unittest.mock import MagicMock
 from unittest.mock import patch
@@ -847,7 +848,6 @@ def test_function_with_int_flag_parameter_with_specific_flags(self):
 
     def test_function_with_int_flag_parameter_invalid_type(self):
         invalid_flag = "not_an_intflag"
-        import pytest
         with nifake.Session('dev1') as session:
             with pytest.raises(TypeError) as exc_info:
                 session.function_with_int_flag_parameter(invalid_flag)
@@ -857,7 +857,6 @@ def test_function_with_int_flag_parameter_invalid_type(self):
     def test_session_write_waveform_numpy_complex64_invalid_dtype(self):
         invalid_waveform_data = numpy.full(10, 1.0 + 1.0j, dtype=numpy.complex128)
         expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=complex64, is complex128"
-        import pytest
         with nifake.Session('dev1') as session:
             with pytest.raises(TypeError) as exc_info:
                 session.write_waveform_numpy_complex64(invalid_waveform_data)
@@ -866,7 +865,6 @@ def test_session_write_waveform_numpy_complex64_invalid_dtype(self):
     def test_session_write_waveform_numpy_complex128_invalid_dtype(self):
         invalid_waveform_data = numpy.full(10, 1.0 + 1.0j, dtype=numpy.complex64)
         expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=complex128, is complex64"
-        import pytest
         with nifake.Session('dev1') as session:
             with pytest.raises(TypeError) as exc_info:
                 session.write_waveform_numpy_complex128(invalid_waveform_data)
@@ -875,7 +873,6 @@ def test_session_write_waveform_numpy_complex128_invalid_dtype(self):
     def test_session_write_waveform_numpy_complex_interleaved_i16_invalid_dtype(self):
         invalid_waveform_data = numpy.full(10, 1.0 + 1.0j, dtype=numpy.complex64)
         expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=int16, is complex64"
-        import pytest
         with nifake.Session('dev1') as session:
             with pytest.raises(TypeError) as exc_info:
                 session.write_waveform_numpy_complex_interleaved_i16(invalid_waveform_data)