Adding unit test

Author: ritaagarwala-sf

tests/test_function_utils.py

@@ -19,21 +19,12 @@
 import sys
 import tempfile
 import textwrap
-from typing import List
 
 import pytest
-from pydantic import BaseModel
 
 from datacustomcode import function_utils
 
 
-class SampleRequest(BaseModel):
-    message: str
-    count: int = 5
-    tags: List[str] = []
-    version: str = "v1"
-
-
 @pytest.fixture
 def sample_entrypoint():
     """Create a temporary entrypoint file with a function."""