Fix snowflake module import (#61)

Fixed some old import references in the snowflake reader and writer modules.

src/koheesio/spark/readers/snowflake.py

@@ -28,6 +28,6 @@
 More detailed class descriptions can be found in the class docstrings.
 """
 
-from koheesio.steps.integrations.snowflake import DbTableQuery, Query, SnowflakeReader
+from koheesio.spark.snowflake import DbTableQuery, Query, SnowflakeReader
 
 __all__ = ["SnowflakeReader", "Query", "DbTableQuery"]

src/koheesio/spark/writers/snowflake.py

@@ -1,5 +1,5 @@
 """This module contains the SnowflakeWriter class, which is used to write data to Snowflake."""
 
-from koheesio.steps.integrations.snowflake import SnowflakeWriter
+from koheesio.spark.snowflake import SnowflakeWriter
 
 __all__ = ["SnowflakeWriter"]

tests/spark/integrations/snowflake/test_snowflake.py

@@ -39,6 +39,11 @@
     "warehouse": "warehouse",
 }
 
+def test_snowflake_module_import():
+    # test that the pass-through imports in the koheesio.spark snowflake modules are working
+    from koheesio.spark.writers import snowflake as snowflake_readers
+    from koheesio.spark.readers import snowflake as snowflake_writers
+
 
 class TestSnowflakeReader:
     reader_options = {"dbtable": "table", **COMMON_OPTIONS}