def test_get_zonecode(self): ''' Test to get current timezone (i.e. PST, MDT, etc) ''' with patch.dict(timezone.__salt__, {'cmd.run': MagicMock(return_value='A')}): self.assertEqual(timezone.get_zonecode(), 'A')