Added test case for mp

ELVIS-Project · Jul 21, 2016 · fd0f7dd · fd0f7dd
1 parent 9ab2f58
commit fd0f7dd
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/vis/tests/test_frequency_experimenter.py b/vis/tests/test_frequency_experimenter.py
@@ -120,6 +120,14 @@ def test_run_3(self):
             else:
                 self.assertEqual(self.freq_d[each], right_df['d'][each])
 
+    def test_mp(self):
+        def fake_func(dataframe, my):
+            return dataframe
+
+        freq = FrequencyExperimenter(self.in_a)
+        result = freq._do_multiprocessing(fake_func, [[2, 1]])
+        self.assertEqual([2], result)
+
 #--------------------------------------------------------------------------------------------------#
 # Definitions                                                                                      #
 #--------------------------------------------------------------------------------------------------#