Check Square Matrix in Python

def main():
	"""
	Prints if a matrix is a square matrix or not.
	"""

	count = int(input("Enter length: "))
	print("Enter matrix: ")
	matrix = read_matrix(count, count)

	if is_square(matrix):
		print("Square Matrix")
	else:
		print("Not a Square Matrix")


def is_square(matrix: list[list[int]]):
	"""
	Checks if a matrix is a square matrix.
	"""

	n = len(matrix)
	for row in matrix:
		if len(row) != n:
			return False
	return True


def read_matrix(row_count: int, col_count: int):
	"""
	Reads a matrix from the console.
	"""

	matrix: list[list[int]] = []
	for _ in range(row_count):
		row = list(map(int, input().split()[:col_count]))
		matrix.append(row)

	return matrix


if __name__ == "__main__":
	main()