make mock import PyVer dependant

Author: Kully

plotly/tests/test_core/test_figure_messages/test_add_traces.py

@@ -1,7 +1,10 @@
 from unittest import TestCase
-from mock import MagicMock
 import plotly.graph_objs as go
 
+if sys.version_info.major == 3 and sys.version_info.minor >= 3:
+    from unittest.mock import MagicMock
+else:
+    from mock import MagicMock
 
 class TestAddTracesMessage(TestCase):
     def setUp(self):

plotly/tests/test_core/test_figure_messages/test_batch_animate.py

@@ -1,7 +1,11 @@
 from unittest import TestCase
-from mock import MagicMock
 import plotly.graph_objs as go
 
+if sys.version_info.major == 3 and sys.version_info.minor >= 3:
+    from unittest.mock import MagicMock
+else:
+    from mock import MagicMock
+
 
 class TestBatchAnimateMessage(TestCase):
     def setUp(self):

plotly/tests/test_core/test_figure_messages/test_move_delete_traces.py

@@ -1,8 +1,13 @@
 from unittest import TestCase
-from mock import MagicMock
 import plotly.graph_objs as go
 from nose.tools import raises
 
+if sys.version_info.major == 3 and sys.version_info.minor >= 3:
+    from unittest.mock import MagicMock
+else:
+    from mock import MagicMock
+
+
 class TestMoveDeleteTracesMessages(TestCase):
     def setUp(self):
         # Construct initial scatter object

plotly/tests/test_core/test_figure_messages/test_on_change.py

@@ -1,8 +1,12 @@
 from unittest import TestCase
-from mock import MagicMock
 import plotly.graph_objs as go
 from nose.tools import raises
 
+if sys.version_info.major == 3 and sys.version_info.minor >= 3:
+    from unittest.mock import MagicMock
+else:
+    from mock import MagicMock
+
 
 class TestOnChangeCallbacks(TestCase):
     def setUp(self):

plotly/tests/test_core/test_figure_messages/test_plotly_relayout.py

@@ -1,8 +1,12 @@
 from unittest import TestCase
-from mock import MagicMock
 from nose.tools import raises
 import plotly.graph_objs as go
 
+if sys.version_info.major == 3 and sys.version_info.minor >= 3:
+    from unittest.mock import MagicMock
+else:
+    from mock import MagicMock
+
 
 class TestRelayoutMessage(TestCase):
 

plotly/tests/test_core/test_figure_messages/test_plotly_restyle.py

@@ -1,9 +1,13 @@
 from unittest import TestCase
-from mock import MagicMock
 from nose.tools import raises
 
 import plotly.graph_objs as go
 
+if sys.version_info.major == 3 and sys.version_info.minor >= 3:
+    from unittest.mock import MagicMock
+else:
+    from mock import MagicMock
+
 
 class TestRestyleMessage(TestCase):
 

plotly/tests/test_core/test_figure_messages/test_plotly_update.py

@@ -1,10 +1,14 @@
 from unittest import TestCase
-from mock import MagicMock
 from nose.tools import raises
 
 import plotly.graph_objs as go
 from plotly.basedatatypes import Undefined
 
+if sys.version_info.major == 3 and sys.version_info.minor >= 3:
+    from unittest.mock import MagicMock
+else:
+    from mock import MagicMock
+
 
 class TestBatchUpdateMessage(TestCase):
     def setUp(self):