Refactor main.py and add unit tests

[sweep-nightly[bot]]

Description

This pull request refactors the main.py file and adds unit tests to the test_main.py file.

Summary

• Refactored main.py to include new functions for initializing a user session, conducting a typing session, and displaying user statistics
• Added unit tests for the new functions in main.py
• Added unit tests for the main function in main.py
• Added unit tests for handling invalid choices in the main function
• Updated test_main.py

Fixes #2.

---

🎉 Latest improvements to Sweep:

• Sweep now uses the rope library to refactor Python! Check out Large Language Models are Bad at Refactoring Code. To have Sweep refactor your code, try sweep: Refactor <your_file>.py!
• Sweep finds and fixes bugs in your code by writing unit tests! Check out Having GPT-4 Iterate on Unit Tests like a Human.

---

💡 To get Sweep to edit this pull request, you can:

• Comment below, and Sweep can edit the entire PR
• Comment on a file, Sweep will only modify the commented file
• Edit the original issue to get Sweep to recreate the PR from scratch

[sweep-nightly[bot]]

Sandbox Executions

• [X] Ran sandbox for tests/test_main.py. ✗

  Sandbox logs for

trunk fmt tests/test_main.py || return 0 1/1 ❌ (`2`)

  FAILURES  
 black  tests/test_main.py  .trunk/out/dQ8OU.yaml
  NOTICES  
 A tool failed to run. You can open the details yaml file for more information.
Checked 1 file
✖ No issues, 1 failure
Run trunk upgrade to upgrade 1 linter
bash: line 1: return: can only `return' from a function or sourced script

• [X] Ran sandbox for tests/test_main.py. ✗

  Sandbox logs for

trunk fmt tests/test_main.py || return 0 1/1 ❌ (`2`)

  FAILURES  
 black  tests/test_main.py  .trunk/out/0cJKn.yaml
  NOTICES  
 A tool failed to run. You can open the details yaml file for more information.
Checked 1 file
✖ No issues, 1 failure
Run trunk upgrade to upgrade 1 linter
bash: line 1: return: can only `return' from a function or sourced script

• [X] Ran sandbox for tests/test_main.py. ✗

  Run `tests/test_main.py` through the sandbox.

• [X] Ran sandbox for tests/test_main.py. ✗

  Sandbox logs for

trunk fmt tests/test_main.py || return 0 1/1 ❌ (`2`)

  FAILURES  
 black  tests/test_main.py  .trunk/out/0cJKn.yaml
  NOTICES  
 A tool failed to run. You can open the details yaml file for more information.
Checked 1 file
✖ No issues, 1 failure
Run trunk upgrade to upgrade 1 linter
bash: line 1: return: can only `return' from a function or sourced script

• [X] Ran sandbox for tests/test_main.py. ✗

  Sandbox logs for

trunk fmt tests/test_main.py || return 0 1/1 ❌ (`2`)

  FAILURES  
 black  tests/test_main.py  .trunk/out/dQ8OU.yaml
  NOTICES  
 A tool failed to run. You can open the details yaml file for more information.
Checked 1 file
✖ No issues, 1 failure
Run trunk upgrade to upgrade 1 linter
bash: line 1: return: can only `return' from a function or sourced script

[sweep-nightly[bot]]

Apply Sweep Rules to your PR?

• [ ] Apply: All docstrings and comments should be up to date.
• [ ] Apply: Code should not contain commented out code blocks.
• [ ] Apply: Variable and function names should be descriptive and follow a consistent naming convention.
• [ ] Apply: Avoid redundant or unused imports and variables.
• [ ] Apply: We import all modules using absolute imports.
• [ ] Apply: All new business logic should have corresponding unit tests in the tests/ directory.
• [ ] Apply: For unit tests we use pytest. If we have class.py, we should have class_test.py in the tests directory.
• [ ] Apply: Any clearly inefficient or repeated code should be optimized or refactored.
• [ ] Apply: Update the README.md file whenever a new feature is added. Make sure to include a description of the feature and any new commands.
• [ ] Apply: Python error logs should use loguru. If they are under an exception, they should use traceback.