Updating the test cases

Author: rahur-NI

generated/nifake/nifake/unit_tests/test_grpc.py

@@ -546,6 +546,31 @@ def test_multiple_arrays_different_size(self):
             data_array=data_array,
         )
 
+    def test_multiple_arrays_different_size_none_input(self):
+        library_func = 'MultipleArraysDifferentSize'
+        response_object = self._set_side_effect(library_func)
+        values_array = [1.1, 2.2, 3.3]
+        interpreter = self._get_initialized_stub_interpreter()
+        assert interpreter.multiple_arrays_different_size(values_array, None) is None  # no outputs
+        self._assert_call(library_func, response_object).assert_called_once_with(
+            vi=GRPC_SESSION_OBJECT_FOR_TEST,
+            values_array=values_array,
+            data_array=None,
+        )
+
+    def test_multiple_arrays_different_size_empty_arrays(self):
+        library_func = 'MultipleArraysDifferentSize'
+        response_object = self._set_side_effect(library_func)
+        values_array = []
+        data_array = []
+        interpreter = self._get_initialized_stub_interpreter()
+        assert interpreter.multiple_arrays_different_size(values_array, data_array) is None  # no outputs
+        self._assert_call(library_func, response_object).assert_called_once_with(
+            vi=GRPC_SESSION_OBJECT_FOR_TEST,
+            values_array=values_array,
+            data_array=data_array,
+        )
+
     def test_multiple_arrays_same_size_wrong_size(self):
         library_func = 'MultipleArraysSameSize'
         # grpc-device server checks this server-side and errors with ::grpc::INVALID_ARGUMENT

generated/nifake/nifake/unit_tests/test_library_interpreter.py

@@ -339,6 +339,33 @@ def test_multiple_arrays_different_size(self):
             _matchers.ViInt32Matcher(len(data_array)),
         )
 
+    def test_multiple_arrays_different_size_none_input(self):
+        self.patched_library.niFake_MultipleArraysDifferentSize.side_effect = self.side_effects_helper.niFake_MultipleArraysDifferentSize
+        values_array = [1.1, 2.2, 3.3]
+        interpreter = self.get_initialized_library_interpreter()
+        interpreter.multiple_arrays_different_size(values_array, None)
+        self.patched_library.niFake_MultipleArraysDifferentSize.assert_called_once_with(
+            _matchers.ViSessionMatcher(SESSION_NUM_FOR_TEST),
+            _matchers.ViReal64BufferMatcher(values_array),
+            _matchers.ViInt32Matcher(len(values_array)),
+            None,
+            _matchers.ViInt32Matcher(0),
+        )
+
+    def test_multiple_arrays_different_size_empty_arrays(self):
+        self.patched_library.niFake_MultipleArraysDifferentSize.side_effect = self.side_effects_helper.niFake_MultipleArraysDifferentSize
+        values_array = []
+        data_array = []
+        interpreter = self.get_initialized_library_interpreter()
+        interpreter.multiple_arrays_different_size(values_array, data_array)
+        self.patched_library.niFake_MultipleArraysDifferentSize.assert_called_once_with(
+            _matchers.ViSessionMatcher(SESSION_NUM_FOR_TEST),
+            _matchers.ViReal64BufferMatcher(values_array),
+            _matchers.ViInt32Matcher(0),
+            _matchers.ViInt32BufferMatcher(data_array),
+            _matchers.ViInt32Matcher(0),
+        )
+
     def test_parameters_are_multiple_types(self):
         self.patched_library.niFake_ParametersAreMultipleTypes.side_effect = self.side_effects_helper.niFake_ParametersAreMultipleTypes
         boolean_val = True

generated/nifake/nifake/unit_tests/test_session.py

@@ -744,6 +744,21 @@ def test_multiple_arrays_different_size(self):
             assert session.multiple_arrays_different_size(values_array, data_array) is None
             self.patched_library_interpreter.multiple_arrays_different_size.assert_called_once_with(values_array, data_array)
 
+    def test_multiple_arrays_different_size_none_input(self):
+        values_array = [1.1, 2.2, 3.3]
+        self.patched_library_interpreter.multiple_arrays_different_size.side_effect = [None]
+        with nifake.Session('dev1') as session:
+            assert session.multiple_arrays_different_size(values_array, None) is None
+            self.patched_library_interpreter.multiple_arrays_different_size.assert_called_once_with(values_array, None)
+
+    def test_multiple_arrays_different_size_empty_arrays(self):
+        values_array = []
+        data_array = []
+        self.patched_library_interpreter.multiple_arrays_different_size.side_effect = [None]
+        with nifake.Session('dev1') as session:
+            assert session.multiple_arrays_different_size(values_array, data_array) is None
+            self.patched_library_interpreter.multiple_arrays_different_size.assert_called_once_with(values_array, data_array)
+
     def test_get_cal_date_time(self):
         month = 12
         day = 30