WIP

test.py

@@ -1,9 +1,97 @@
-def main():
-    print("hello world")
-    x = 50
+import random
+from typing import Any, Dict, List
 
-    for i in range(x):
-        print(i)
+
+class Employee:
+    def __init__(self, name: str, age: int, skills: List[str]):
+        self.name = name
+        self.age = age
+        self.skills = skills
+        self.projects: Dict[str, Dict[str, Any]] = {}
+
+    def add_project(self, project_name: str, details: Dict[str, Any]):
+        self.projects[project_name] = details
+
+    def __repr__(self):
+        return f"<Employee {self.name}, age={self.age}, skills={self.skills}>"
+
+
+class Company:
+    def __init__(self, name: str):
+        self.name = name
+        self.employees: List[Employee] = []
+
+    def hire_employee(self, employee: Employee):
+        self.employees.append(employee)
+
+    def list_employees(self):
+        return [e.name for e in self.employees]
+
+    def employee_summary(self) -> Dict[str, Any]:
+        summary = {}
+        for e in self.employees:
+            summary[e.name] = {"age": e.age, "skills": e.skills, "projects": e.projects}
+        return summary
+
+
+def random_employee() -> Employee:
+    names = ["Alice", "Bob", "Charlie", "Diana", "Eve", "Frank"]
+    skills_pool = ["Python", "Go", "Rust", "JavaScript", "SQL", "Docker"]
+    name = random.choice(names)
+    age = random.randint(22, 45)
+    skills = random.sample(skills_pool, k=random.randint(1, 4))
+    return Employee(name, age, skills)
+
+
+def generate_company(name: str, num_employees: int) -> Company:
+    company = Company(name)
+    for _ in range(num_employees):
+        emp = random_employee()
+        # assign some random projects
+        for pid in range(random.randint(1, 3)):
+            proj_name = f"Project-{pid+1}"
+            emp.add_project(
+                proj_name,
+                {
+                    "budget": random.randint(1000, 10000),
+                    "deadline": f"2026-0{random.randint(1,9)}-{random.randint(10,28)}",
+                    "tasks": [f"Task-{i+1}" for i in range(random.randint(1, 5))],
+                },
+            )
+        company.hire_employee(emp)
+    return company
+
+
+def main():
+    my_company = generate_company("TechCorp", 5)
+
+    print("All employees:")
+    for emp in my_company.employees:
+        print(emp)
+
+    summary = my_company.employee_summary()
+    print("\nEmployee Summary Dictionary:")
+    for name, info in summary.items():
+        print(f"{name}: {info}")
+
+    # Access nested structures
+    print("\nAccessing nested info:")
+    for emp in my_company.employees:
+        for proj, details in emp.projects.items():
+            print(
+                f"{emp.name}'s {proj} budget is {details['budget']} with tasks {details['tasks']}"
+            )
+
+    # Filter employees with Python skill
+    pythonistas = [e.name for e in my_company.employees if "Python" in e.skills]
+    print(f"\nEmployees with Python skill: {pythonistas}")
+
+    # Update project budget for first employee
+    if my_company.employees:
+        first = my_company.employees[0]
+        for proj in first.projects:
+            first.projects[proj]["budget"] += 500
+        print(f"\nUpdated {first.name}'s projects: {first.projects}")
 
 
 if __name__ == "__main__":