def test_version(self): ''' Test for Return the cassandra version ''' mock = MagicMock(return_value='A') with patch.object(cassandra, '_nodetool', mock): self.assertEqual(cassandra.version(), 'A')
def test_version(self): """ Test for Return the cassandra version """ mock = MagicMock(return_value="A") with patch.object(cassandra, "_nodetool", mock): self.assertEqual(cassandra.version(), "A")