def test_version(self):
"""
Test for Return version from hadoop version
"""
mock = MagicMock(return_value="A \nB \n")
with patch.dict(hadoop.__salt__, {"cmd.run": mock}):
self.assertEqual(hadoop.version(), "B")
def test_version(self):
'''
Test for Return version from hadoop version
'''
mock = MagicMock(return_value="A \nB \n")
with patch.dict(hadoop.__salt__, {'cmd.run': mock}):
self.assertEqual(hadoop.version(), 'B')
def test_version(self):
'''
Test for Return version from hadoop version
'''
mock = MagicMock(return_value="A \nB \n")
with patch.dict(hadoop.__salt__, {'cmd.run': mock}):
self.assertEqual(hadoop.version(), 'B')