List Comprehension and Functions – Python Code Typing Test

Demonstrates Python's list comprehension, type hints, and function definitions with docstrings.

import math
from typing import List, Optional

def fibonacci(n: int) -> List[int]:
	"""Generate fibonacci sequence up to n numbers"""
	if n <= 0:
		return []
	elif n == 1:
		return [0]

	fib = [0, 1]
	for i in range(2, n):
		fib.append(fib[i-1] + fib[i-2])
	return fib

def find_primes(limit: int) -> List[int]:
	"""Find all prime numbers up to limit using sieve"""
	primes = []
	sieve = [True] * (limit + 1)

	for i in range(2, int(math.sqrt(limit)) + 1):
		if sieve[i]:
		for j in range(i * i, limit + 1, i):
		sieve[j] = False

	return [i for i in range(2, limit + 1) if sieve[i]]

# Main execution
if __name__ == "__main__":
	fib_sequence = fibonacci(10)
	print(f"Fibonacci: {fib_sequence}")

	primes = find_primes(30)
	print(f"Primes up to 30: {primes}")

	# List comprehension example
	squares = [x**2 for x in range(1, 11) if x % 2 == 0]
	print(f"Even squares: {squares}")