Fix mypy errors in test helper

Author: fern-support

tests/integration/helpers.py

@@ -9,9 +9,7 @@
 from square.types.address import Address
 from square.types.catalog_item import CatalogItem
 from square.types.catalog_item_variation import CatalogItemVariation
-from square.types.catalog_object import CatalogObject
-from square.types.catalog_object_item import CatalogObjectItem
-from square.types.catalog_object_item_variation import CatalogObjectItemVariation
+from square.types.catalog_object import CatalogObject, CatalogObject_Item, CatalogObject_ItemVariation
 from square.types.location import Location
 from square.types.money import Money
 
@@ -47,7 +45,7 @@ def __init__(self):
 
     def variations(self) -> List[CatalogObject]:
         return [
-            CatalogObjectItemVariation(
+            CatalogObject_ItemVariation(
                 type="ITEM_VARIATION",
                 id="#variation" + str(uuid.uuid4()),
                 present_at_all_locations=True,
@@ -69,7 +67,7 @@ def new_test_money(amount: int) -> Money:
 
 
 def create_test_catalog_item(opts: CreateCatalogItemOptions) -> CatalogObject:
-    return CatalogObjectItem(
+    return CatalogObject_Item(
         type="ITEM",
         id="#" + str(uuid.uuid4()),
         present_at_all_locations=True,

tests/integration/test_inventory.py

@@ -7,8 +7,8 @@
 from square.core.api_error import ApiError
 from square.requests.catalog_item import CatalogItemParams
 from square.requests.catalog_item_variation import CatalogItemVariationParams
-from square.requests.catalog_object_item_variation import (
-    CatalogObjectItemVariationParams,
+from square.requests.catalog_object import (
+    CatalogObject_ItemVariationParams,
 )
 from square.types.catalog_item import CatalogItem
 from square.types.catalog_object_item import CatalogObjectItem
@@ -49,7 +49,7 @@ def create_catalog_item_variation() -> str:
         "price_money": {"amount": 100, "currency": "USD"},
     }
 
-    variation: CatalogObjectItemVariationParams = {
+    variation: CatalogObject_ItemVariationParams = {
         "type": "ITEM_VARIATION",
         "id": "#colombian-coffee",
         "present_at_all_locations": True,