diff --git a/kittycad/client_test.py b/kittycad/client_test.py index 5eed185e8..9bb19dce1 100644 --- a/kittycad/client_test.py +++ b/kittycad/client_test.py @@ -3,8 +3,8 @@ import pytest import asyncio from .client import ClientFromEnv -from .models import FileConversion, FileOutputFormat, FileSourceFormat, User, Pong, APICallStatus -from .api.file import create_file_conversion_with_base64_helper +from .models import FileConversion, FileOutputFormat, FileSourceFormat, User, Pong, APICallStatus, FileMass, FileVolume +from .api.file import create_file_conversion_with_base64_helper, create_file_mass, create_file_volume from .api.meta import ping from .api.users import get_user_self @@ -106,3 +106,56 @@ async def test_file_convert_stl_async(): assert fc.id is not None print(f"FileConversion: {fc}") + +def test_file_mass(): + # Create our client. + client = ClientFromEnv() + + dir_path = os.path.dirname(os.path.realpath(__file__)) + file = open(os.path.join(dir_path, "../assets/testing.stl"), "rb") + content = file.read() + file.close() + + # Get the fc. + fm: FileMass = create_file_mass.sync( + client=client, + body=content, + src_format=FileSourceFormat.STL, + material_density=1.0) + + assert fm is not None + + print(f"FileConversion: {fm}") + + assert fm.id is not None + assert fm.mass is not None + + assert fm.status == APICallStatus.COMPLETED + + print(f"FileConversion: {fm}") + +def test_file_volume(): + # Create our client. + client = ClientFromEnv() + + dir_path = os.path.dirname(os.path.realpath(__file__)) + file = open(os.path.join(dir_path, "../assets/testing.stl"), "rb") + content = file.read() + file.close() + + # Get the fc. + fv: FileVolume = create_file_volume.sync( + client=client, + body=content, + src_format=FileSourceFormat.STL) + + assert fv is not None + + print(f"FileConversion: {fv}") + + assert fv.id is not None + assert fv.volume is not None + + assert fv.status == APICallStatus.COMPLETED + + print(f"FileConversion: {fv}")