Updating the test cases

Author: rahulr-NI

generated/nifake/nifake/unit_tests/test_library_interpreter.py

@@ -865,30 +865,6 @@ def test_write_waveform_numpy_complex128_valid_input(self):
             _matchers.NIComplexNumberPointerMatcher(waveform_data_pointer, number_of_samples)
         )
 
-    def test_write_waveform_numpy_complex64_invalid_input(self):
-        import numpy as np
-
-        invalid_waveform_data = np.full(1000, 0.707 + 0.707j, dtype=np.complex128)
-        expected_error_message = "Invalid waveform data provided. Expected a non-empty array of complex64."
-        interpreter = self.get_initialized_library_interpreter()
-        self.patched_library.niFake_WriteWaveformComplexF32.side_effect = ValueError("Invalid waveform data provided. Expected a non-empty array of complex64.")
-        with pytest.raises(ValueError) as exc_info:
-            interpreter.write_waveform_complex_f32(invalid_waveform_data)
-
-        assert str(exc_info.value) == expected_error_message
-
-    def test_write_interleaved_complexi16_invalid_input(self):
-        import numpy as np
-
-        invalid_waveform_data = np.array([], dtype=np.complex64)
-        expected_error_message = "Invalid waveform data provided. Expected a non-empty array of Int16."
-
-        interpreter = self.get_initialized_library_interpreter()
-        self.patched_library.niFake_WriteWaveformComplexI16.side_effect = ValueError("Invalid waveform data provided. Expected a non-empty array of Int16.")
-        with pytest.raises(ValueError) as exc_info:
-            interpreter.write_waveform_complex_i16(invalid_waveform_data)
-        assert str(exc_info.value) == expected_error_message
-
     def test_write_waveform_complexi16_valid_input(self):
         import ctypes
         import numpy as np

generated/nifake/nifake/unit_tests/test_session.py

@@ -926,6 +926,26 @@ def test_export_attribute_configuration_buffer(self):
             assert actual_configuration == bytes(expected_buffer)
         self.patched_grpc_interpreter.export_attribute_configuration_buffer.assert_called_once_with()
 
+    def test_session_write_waveform_complex_f32_invalid_dtype(self):
+        import numpy as np
+        invalid_waveform_data = np.full(10, 1.0 + 1.0j, dtype=np.complex128)
+        expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=complex64, is complex128"
+        import pytest
+        with nifake.Session('dev1', grpc_options=nifake.GrpcSessionOptions(object(), '')) as session:
+            with pytest.raises(TypeError) as exc_info:
+                session.write_waveform_complex_f32(invalid_waveform_data)
+            assert str(exc_info.value) == expected_error_message
+
+    def test_session_write_waveform_complex_i16_invalid_dtype(self):
+        import numpy as np
+        invalid_waveform_data = np.full(10, 1.0 + 1.0j, dtype=np.complex64)
+        expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=int16, is complex64"
+        import pytest
+        with nifake.Session('dev1', grpc_options=nifake.GrpcSessionOptions(object(), '')) as session:
+            with pytest.raises(TypeError) as exc_info:
+                session.write_waveform_complex_i16(invalid_waveform_data)
+            assert str(exc_info.value) == expected_error_message
+
     # Attributes
 
     def test_get_attribute_int32(self):

src/nifake/unit_tests/test_library_interpreter.py

@@ -865,30 +865,6 @@ def test_write_waveform_numpy_complex128_valid_input(self):
             _matchers.NIComplexNumberPointerMatcher(waveform_data_pointer, number_of_samples)
         )
 
-    def test_write_waveform_numpy_complex64_invalid_input(self):
-        import numpy as np
-
-        invalid_waveform_data = np.full(1000, 0.707 + 0.707j, dtype=np.complex128)
-        expected_error_message = "Invalid waveform data provided. Expected a non-empty array of complex64."
-        interpreter = self.get_initialized_library_interpreter()
-        self.patched_library.niFake_WriteWaveformComplexF32.side_effect = ValueError("Invalid waveform data provided. Expected a non-empty array of complex64.")
-        with pytest.raises(ValueError) as exc_info:
-            interpreter.write_waveform_complex_f32(invalid_waveform_data)
-
-        assert str(exc_info.value) == expected_error_message
-
-    def test_write_interleaved_complexi16_invalid_input(self):
-        import numpy as np
-
-        invalid_waveform_data = np.array([], dtype=np.complex64)
-        expected_error_message = "Invalid waveform data provided. Expected a non-empty array of Int16."
-
-        interpreter = self.get_initialized_library_interpreter()
-        self.patched_library.niFake_WriteWaveformComplexI16.side_effect = ValueError("Invalid waveform data provided. Expected a non-empty array of Int16.")
-        with pytest.raises(ValueError) as exc_info:
-            interpreter.write_waveform_complex_i16(invalid_waveform_data)
-        assert str(exc_info.value) == expected_error_message
-
     def test_write_waveform_complexi16_valid_input(self):
         import ctypes
         import numpy as np

src/nifake/unit_tests/test_session.py

@@ -926,6 +926,26 @@ def test_export_attribute_configuration_buffer(self):
             assert actual_configuration == bytes(expected_buffer)
         self.patched_grpc_interpreter.export_attribute_configuration_buffer.assert_called_once_with()
 
+    def test_session_write_waveform_complex_f32_invalid_dtype(self):
+        import numpy as np
+        invalid_waveform_data = np.full(10, 1.0 + 1.0j, dtype=np.complex128)
+        expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=complex64, is complex128"
+        import pytest
+        with nifake.Session('dev1', grpc_options=nifake.GrpcSessionOptions(object(), '')) as session:
+            with pytest.raises(TypeError) as exc_info:
+                session.write_waveform_complex_f32(invalid_waveform_data)
+            assert str(exc_info.value) == expected_error_message
+
+    def test_session_write_waveform_complex_i16_invalid_dtype(self):
+        import numpy as np
+        invalid_waveform_data = np.full(10, 1.0 + 1.0j, dtype=np.complex64)
+        expected_error_message = "waveform_data_array must be numpy.ndarray of dtype=int16, is complex64"
+        import pytest
+        with nifake.Session('dev1', grpc_options=nifake.GrpcSessionOptions(object(), '')) as session:
+            with pytest.raises(TypeError) as exc_info:
+                session.write_waveform_complex_i16(invalid_waveform_data)
+            assert str(exc_info.value) == expected_error_message
+
     # Attributes
 
     def test_get_attribute_int32(self):