old_contents: from click.testing import CliRunner
  import unittest
  
  from mock import patch
  from polyaxon_cli.cli.auth import login, logout, whoami
  
  
  class TestAuthClient(unittest.TestCase):
      def setUp(self):
          self.runner = CliRunner()
  
      @patch('polyaxon_cli.managers.auth.AuthConfigManager.purge')
      def test_logout(self, purge_patch):
          self.runner.invoke(logout)
          assert purge_patch.call_count == 1
  
      @patch('polyaxon_client.auth.AuthClient.get_user')
      def test_whoami(self, purge_patch):
          self.runner.invoke(whoami)
          assert purge_patch.call_count == 1
  
new_contents: from click.testing import CliRunner
  import unittest
  
  from mock import patch
  from polyaxon_cli.cli.auth import logout, whoami
  
  
  class TestAuth(unittest.TestCase):
      def setUp(self):
          self.runner = CliRunner()
  
      @patch('polyaxon_cli.managers.auth.AuthConfigManager.purge')
      def test_logout(self, purge_patch):
          self.runner.invoke(logout)
          assert purge_patch.call_count == 1
  
      @patch('polyaxon_client.auth.AuthClient.get_user')
      def test_whoami(self, purge_patch):
          self.runner.invoke(whoami)
          assert purge_patch.call_count == 1
  
subject: Update auth tests