Implement Perfect Square Sum Calculation Function

perfect_square_sum.py

@@ -0,0 +1,33 @@
+def is_perfect_square(n: int) -> bool:
+    """
+    Check if a number is a perfect square.
+
+    :param n: The number to check.
+    :return: True if the number is a perfect square, False otherwise.
+    """
+    sqrt = n ** 0.5
+    return sqrt.is_integer()
+
+
+def perfect_square_sum(input_set: set[int]) -> int:
+    """
+    Calculate the sum of all perfect squares that can be formed from the integers in the set.
+
+    :param input_set: A set of positive integers.
+    :return: The sum of perfect squares that can be formed from the input set.
+    """
+    if not input_set:
+        return 0
+
+    perfect_square_sum = 0
+    for num in input_set:
+        if is_perfect_square(num):
+            perfect_square_sum += num
+
+    return perfect_square_sum
+
+
+if __name__ == "__main__":
+    import doctest
+
+    doctest.testmod()