User-defined functions with Snowpark in Snowflake - Practice Problems & Coding Challenges

Challenge - 5 Problems

❓ service_behavior

intermediate

What is the output of this Snowpark UDF execution?

Given a Snowflake Snowpark UDF that doubles an integer input, what will be the output of calling it with input 5?

Snowflake

from snowflake.snowpark.functions import udf

@udf
def double_value(x: int) -> int:
    return x * 2

result = double_value(5).collect()[0][0]

A 25

B 5

C 10

D None

💡 Hint

Think about what doubling 5 means mathematically.

The UDF multiplies the input by 2, so 5 * 2 = 10.

❓ Configuration

intermediate

Which option correctly registers a Python UDF in Snowflake using Snowpark?

Select the code snippet that correctly registers a Python UDF named add_one that adds 1 to an integer input.

A session.udf.add('add_one', lambda x: x + 1, IntegerType())

B session.udf.register(lambda x: x + 1, return_type=IntegerType(), input_types=[IntegerType()], name='add_one')

C session.create_udf('add_one', lambda x: x + 1, IntegerType())

D session.udf.create('add_one', lambda x: x + 1, IntegerType())

💡 Hint

Look for the method that registers a UDF with name, input, and return types.

The session.udf.register method is the correct way to register a Python UDF with specified input and return types.

❓ Architecture

advanced

Which Snowpark UDF design best supports processing large datasets efficiently?

Choose the UDF design that optimizes performance when applied to large Snowflake tables.

A A vectorized UDF that processes batches of rows in parallel

B A scalar UDF that processes one row at a time without vectorization

C A UDF that writes intermediate results to external storage during execution

D A UDF that uses global variables to store state across calls

💡 Hint

Think about how batch processing can improve speed over row-by-row processing.

Vectorized UDFs process multiple rows at once, reducing overhead and improving performance on large datasets.

❓ security

advanced

What is the best practice to secure Snowpark UDFs that access sensitive data?

Which option ensures that a Snowpark UDF accessing sensitive data follows security best practices?

A Grant the UDF owner minimal privileges and use role-based access control

B Embed credentials inside the UDF code for direct database access

C Allow public access to the UDF for easier sharing

D Disable all logging to hide UDF activity

💡 Hint

Consider how to limit access and track usage securely.

Using minimal privileges and role-based access control limits exposure and follows the principle of least privilege.

🧠 Conceptual

expert

What error will this Snowpark UDF raise when called with a string input?

Consider this UDF that expects an integer input but is called with a string:

from snowflake.snowpark.functions import udf

@udf
def square(x: int) -> int:
    return x * x

result = square('text').collect()[0][0]

What error occurs?

A No error, returns 'texttext'

B ValueError: invalid literal for int() with base 10: 'text'

C SnowparkUDFExecutionError: Input type mismatch

D TypeError: unsupported operand type(s) for *: 'str' and 'str'

💡 Hint

Think about what happens when multiplying a string by a string in Python.

Multiplying a string by a string is not supported in Python, causing a TypeError.

Practice

1. What is the main purpose of user-defined functions (UDFs) in Snowflake using Snowpark?

easy

A. To manage user permissions and roles

B. To create new database tables automatically

C. To run custom code inside Snowflake and reuse logic

D. To monitor Snowflake account usage

Solution

1. Step 1: Understand what UDFs do

   User-defined functions let you write your own code to run inside Snowflake, which helps reuse logic.

2. Step 2: Identify the main purpose

   UDFs simplify queries by encapsulating custom logic, not managing tables, permissions, or monitoring.

3. Final Answer:

   To run custom code inside Snowflake and reuse logic -> Option C

4. Quick Check:

   UDFs = Custom code reuse [OK]

Hint: UDFs run your own code inside Snowflake [OK]

Common Mistakes:

• Thinking UDFs create tables
• Confusing UDFs with security roles
• Assuming UDFs monitor usage

2. Which of the following is the correct way to define a simple Python UDF in Snowpark that adds 10 to an integer input?

easy

A. def add_ten(x): return x + 10

B. def add_ten(x): x + 10

C. def add_ten(x): print(x + 10)

D. def add_ten(x): return x - 10

Solution

1. Step 1: Check function syntax

   A Python function must use 'return' to output a value; def add_ten(x): return x + 10 correctly returns x + 10.

2. Step 2: Verify logic correctness

   def add_ten(x): return x + 10 adds 10 to input x, matching the requirement; others either miss return or subtract.

3. Final Answer:

   def add_ten(x): return x + 10 -> Option A

4. Quick Check:

   Return correct value with 'return' keyword [OK]

Hint: Always use 'return' to output from Python UDFs [OK]

Common Mistakes:

• Omitting the return statement
• Using print instead of return
• Returning wrong calculation

3. Given this Snowpark Python UDF code:

def multiply_by_two(x):
    return x * 2

result = multiply_by_two(5)

What is the value of result?

medium

A. 5

B. None

C. 25

D. 10

Solution

1. Step 1: Understand the function logic

   The function multiplies input x by 2 and returns the result.

2. Step 2: Calculate the function output for input 5

   5 multiplied by 2 equals 10, so result is 10.

3. Final Answer:

   10 -> Option D

4. Quick Check:

   5 * 2 = 10 [OK]

Hint: Multiply input by 2 to get output [OK]

Common Mistakes:

• Confusing multiplication with addition
• Assuming function returns None without return
• Mistaking input for output

4. You wrote this Snowpark Python UDF:

def square(x):
    x * x

When you run it, you get no output. What is the problem?

medium

A. Function name is invalid

B. Missing return statement to output the result

C. Syntax error due to missing colon

D. Input parameter x is not defined

Solution

1. Step 1: Check function syntax

   The function lacks a 'return' statement, so it does not output any value.

2. Step 2: Understand Python function behavior

   Without 'return', the function returns None by default, causing no output.

3. Final Answer:

   Missing return statement to output the result -> Option B

4. Quick Check:

   Functions need 'return' to output values [OK]

Hint: Always include 'return' to output from functions [OK]

Common Mistakes:

• Assuming last expression auto-returns
• Thinking function name causes error
• Confusing missing colon with missing return

5. You want to create a Snowpark Python UDF that returns the length of a string only if it is not empty; otherwise, it returns 0. Which function correctly implements this?

hard

A. def str_length(s):
if s:
return len(s)
else:
return 0

B. def str_length(s):
return len(s) if s != '' else 0

C. def str_length(s):
if s == '':
return len(s)
else:
return 0

D. def str_length(s):
return len(s) or 0

Solution

1. Step 1: Understand the condition for empty string

   Empty strings are falsy in Python, so 'if s:' checks if string is not empty.

2. Step 2: Verify correct return values

   If string is not empty, return its length; else return 0. def str_length(s): if s: return len(s) else: return 0 matches this logic exactly.

3. Final Answer:

   def str_length(s): if s: return len(s) else: return 0 -> Option A

4. Quick Check:

   Check string truthiness to decide length [OK]

Hint: Use 'if s:' to check non-empty string [OK]

Common Mistakes:

• Checking 's != \"\"' crashes on None
• Returning length for empty string incorrectly
• Using 'or' without proper condition