snowpark-connect 0.23.0__py3-none-any.whl → 0.25.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of snowpark-connect might be problematic. Click here for more details.
- snowflake/snowpark_connect/column_name_handler.py +116 -4
- snowflake/snowpark_connect/config.py +13 -0
- snowflake/snowpark_connect/constants.py +0 -29
- snowflake/snowpark_connect/dataframe_container.py +6 -0
- snowflake/snowpark_connect/execute_plan/map_execution_command.py +56 -1
- snowflake/snowpark_connect/expression/function_defaults.py +207 -0
- snowflake/snowpark_connect/expression/literal.py +18 -2
- snowflake/snowpark_connect/expression/map_cast.py +5 -8
- snowflake/snowpark_connect/expression/map_expression.py +10 -1
- snowflake/snowpark_connect/expression/map_extension.py +12 -2
- snowflake/snowpark_connect/expression/map_sql_expression.py +23 -1
- snowflake/snowpark_connect/expression/map_udf.py +26 -8
- snowflake/snowpark_connect/expression/map_unresolved_attribute.py +199 -15
- snowflake/snowpark_connect/expression/map_unresolved_extract_value.py +44 -16
- snowflake/snowpark_connect/expression/map_unresolved_function.py +836 -365
- snowflake/snowpark_connect/expression/map_unresolved_star.py +3 -2
- snowflake/snowpark_connect/hidden_column.py +39 -0
- snowflake/snowpark_connect/includes/jars/hadoop-client-api-trimmed-3.3.4.jar +0 -0
- snowflake/snowpark_connect/includes/jars/{hadoop-client-api-3.3.4.jar → spark-connect-client-jvm_2.12-3.5.6.jar} +0 -0
- snowflake/snowpark_connect/relation/map_column_ops.py +18 -36
- snowflake/snowpark_connect/relation/map_extension.py +56 -15
- snowflake/snowpark_connect/relation/map_join.py +258 -62
- snowflake/snowpark_connect/relation/map_row_ops.py +2 -29
- snowflake/snowpark_connect/relation/map_sql.py +88 -11
- snowflake/snowpark_connect/relation/map_udtf.py +4 -2
- snowflake/snowpark_connect/relation/read/map_read.py +3 -3
- snowflake/snowpark_connect/relation/read/map_read_jdbc.py +1 -1
- snowflake/snowpark_connect/relation/read/map_read_json.py +8 -1
- snowflake/snowpark_connect/relation/read/map_read_table.py +1 -9
- snowflake/snowpark_connect/relation/read/reader_config.py +3 -1
- snowflake/snowpark_connect/relation/read/utils.py +6 -7
- snowflake/snowpark_connect/relation/utils.py +1 -170
- snowflake/snowpark_connect/relation/write/map_write.py +62 -53
- snowflake/snowpark_connect/resources_initializer.py +29 -1
- snowflake/snowpark_connect/server.py +18 -3
- snowflake/snowpark_connect/type_mapping.py +29 -25
- snowflake/snowpark_connect/typed_column.py +14 -0
- snowflake/snowpark_connect/utils/artifacts.py +23 -0
- snowflake/snowpark_connect/utils/context.py +6 -1
- snowflake/snowpark_connect/utils/scala_udf_utils.py +588 -0
- snowflake/snowpark_connect/utils/telemetry.py +6 -17
- snowflake/snowpark_connect/utils/udf_helper.py +2 -0
- snowflake/snowpark_connect/utils/udf_utils.py +38 -7
- snowflake/snowpark_connect/utils/udtf_utils.py +17 -3
- snowflake/snowpark_connect/version.py +1 -1
- {snowpark_connect-0.23.0.dist-info → snowpark_connect-0.25.0.dist-info}/METADATA +1 -1
- snowpark_connect-0.25.0.dist-info/RECORD +477 -0
- snowflake/snowpark_connect/includes/jars/scala-compiler-2.12.18.jar +0 -0
- snowflake/snowpark_connect/includes/jars/spark-kubernetes_2.12-3.5.6.jar +0 -0
- snowflake/snowpark_connect/includes/jars/spark-mllib_2.12-3.5.6.jar +0 -0
- snowflake/snowpark_connect/includes/jars/spark-streaming_2.12-3.5.6.jar +0 -0
- snowflake/snowpark_connect/includes/python/pyspark/errors/tests/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/errors/tests/test_errors.py +0 -60
- snowflake/snowpark_connect/includes/python/pyspark/ml/deepspeed/tests/test_deepspeed_distributor.py +0 -306
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_classification.py +0 -53
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_evaluation.py +0 -50
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_feature.py +0 -43
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_function.py +0 -114
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_pipeline.py +0 -47
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_summarizer.py +0 -43
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_tuning.py +0 -46
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_legacy_mode_classification.py +0 -238
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_legacy_mode_evaluation.py +0 -194
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_legacy_mode_feature.py +0 -156
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_legacy_mode_pipeline.py +0 -184
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_legacy_mode_summarizer.py +0 -78
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_legacy_mode_tuning.py +0 -292
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_parity_torch_data_loader.py +0 -50
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_parity_torch_distributor.py +0 -152
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_algorithms.py +0 -456
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_base.py +0 -96
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_dl_util.py +0 -186
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_evaluation.py +0 -77
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_feature.py +0 -401
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_functions.py +0 -528
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_image.py +0 -82
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_linalg.py +0 -409
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_model_cache.py +0 -55
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_param.py +0 -441
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_persistence.py +0 -546
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_pipeline.py +0 -71
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_stat.py +0 -52
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_training_summary.py +0 -494
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_util.py +0 -85
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/test_wrapper.py +0 -138
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/tuning/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/tuning/test_cv_io_basic.py +0 -151
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/tuning/test_cv_io_nested.py +0 -97
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/tuning/test_cv_io_pipeline.py +0 -143
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/tuning/test_tuning.py +0 -551
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/tuning/test_tvs_io_basic.py +0 -137
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/tuning/test_tvs_io_nested.py +0 -96
- snowflake/snowpark_connect/includes/python/pyspark/ml/tests/tuning/test_tvs_io_pipeline.py +0 -142
- snowflake/snowpark_connect/includes/python/pyspark/ml/torch/tests/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/ml/torch/tests/test_data_loader.py +0 -137
- snowflake/snowpark_connect/includes/python/pyspark/ml/torch/tests/test_distributor.py +0 -561
- snowflake/snowpark_connect/includes/python/pyspark/ml/torch/tests/test_log_communication.py +0 -172
- snowflake/snowpark_connect/includes/python/pyspark/mllib/tests/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/mllib/tests/test_algorithms.py +0 -353
- snowflake/snowpark_connect/includes/python/pyspark/mllib/tests/test_feature.py +0 -192
- snowflake/snowpark_connect/includes/python/pyspark/mllib/tests/test_linalg.py +0 -680
- snowflake/snowpark_connect/includes/python/pyspark/mllib/tests/test_stat.py +0 -206
- snowflake/snowpark_connect/includes/python/pyspark/mllib/tests/test_streaming_algorithms.py +0 -471
- snowflake/snowpark_connect/includes/python/pyspark/mllib/tests/test_util.py +0 -108
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_any_all.py +0 -177
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_apply_func.py +0 -575
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_binary_ops.py +0 -235
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_combine.py +0 -653
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_compute.py +0 -463
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_corrwith.py +0 -86
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_cov.py +0 -151
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_cumulative.py +0 -139
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_describe.py +0 -458
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_eval.py +0 -86
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_melt.py +0 -202
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_missing_data.py +0 -520
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/computation/test_pivot.py +0 -361
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_any_all.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_apply_func.py +0 -42
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_binary_ops.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_combine.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_compute.py +0 -60
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_corrwith.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_cov.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_cumulative.py +0 -90
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_describe.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_eval.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_melt.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_missing_data.py +0 -42
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/computation/test_parity_pivot.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_base.py +0 -36
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_binary_ops.py +0 -42
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_boolean_ops.py +0 -47
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_categorical_ops.py +0 -55
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_complex_ops.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_date_ops.py +0 -47
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_datetime_ops.py +0 -47
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_null_ops.py +0 -42
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_num_arithmetic.py +0 -43
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_num_ops.py +0 -47
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_num_reverse.py +0 -43
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_string_ops.py +0 -47
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_timedelta_ops.py +0 -47
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/test_parity_udt_ops.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/data_type_ops/testing_utils.py +0 -226
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/test_parity_align.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/test_parity_basic_slow.py +0 -55
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/test_parity_cov_corrwith.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/test_parity_dot_frame.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/test_parity_dot_series.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/test_parity_index.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/test_parity_series.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/test_parity_setitem_frame.py +0 -43
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/diff_frames_ops/test_parity_setitem_series.py +0 -43
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/test_parity_attrs.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/test_parity_constructor.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/test_parity_conversion.py +0 -42
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/test_parity_reindexing.py +0 -42
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/test_parity_reshaping.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/test_parity_spark.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/test_parity_take.py +0 -42
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/test_parity_time_series.py +0 -48
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/frame/test_parity_truncate.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_aggregate.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_apply_func.py +0 -41
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_cumulative.py +0 -67
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_describe.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_groupby.py +0 -55
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_head_tail.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_index.py +0 -38
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_missing_data.py +0 -55
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_split_apply.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/groupby/test_parity_stat.py +0 -38
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/test_parity_align.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/test_parity_base.py +0 -50
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/test_parity_category.py +0 -73
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/test_parity_datetime.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/test_parity_indexing.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/test_parity_reindex.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/test_parity_rename.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/test_parity_reset_index.py +0 -48
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/indexes/test_parity_timedelta.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/io/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/io/test_parity_io.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/plot/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/plot/test_parity_frame_plot.py +0 -45
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/plot/test_parity_frame_plot_matplotlib.py +0 -45
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/plot/test_parity_frame_plot_plotly.py +0 -49
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/plot/test_parity_series_plot.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/plot/test_parity_series_plot_matplotlib.py +0 -53
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/plot/test_parity_series_plot_plotly.py +0 -45
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_all_any.py +0 -38
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_arg_ops.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_as_of.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_as_type.py +0 -38
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_compute.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_conversion.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_cumulative.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_index.py +0 -38
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_missing_data.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_series.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_sort.py +0 -38
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/series/test_parity_stat.py +0 -38
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_categorical.py +0 -66
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_config.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_csv.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_dataframe_conversion.py +0 -42
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_dataframe_spark_io.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_default_index.py +0 -49
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_ewm.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_expanding.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_extension.py +0 -49
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_frame_spark.py +0 -53
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_generic_functions.py +0 -43
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_indexing.py +0 -49
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_indexops_spark.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_internal.py +0 -41
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_namespace.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_numpy_compat.py +0 -60
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_ops_on_diff_frames.py +0 -48
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_ops_on_diff_frames_groupby.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_ops_on_diff_frames_groupby_expanding.py +0 -44
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_ops_on_diff_frames_groupby_rolling.py +0 -84
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_repr.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_resample.py +0 -45
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_reshape.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_rolling.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_scalars.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_series_conversion.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_series_datetime.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_series_string.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_spark_functions.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_sql.py +0 -43
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_stats.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_typedef.py +0 -36
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_utils.py +0 -37
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/connect/test_parity_window.py +0 -39
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_base.py +0 -107
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_binary_ops.py +0 -224
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_boolean_ops.py +0 -825
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_categorical_ops.py +0 -562
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_complex_ops.py +0 -368
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_date_ops.py +0 -257
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_datetime_ops.py +0 -260
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_null_ops.py +0 -178
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_num_arithmetic.py +0 -184
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_num_ops.py +0 -497
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_num_reverse.py +0 -140
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_string_ops.py +0 -354
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_timedelta_ops.py +0 -219
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_udt_ops.py +0 -192
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/testing_utils.py +0 -228
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/test_align.py +0 -118
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/test_basic_slow.py +0 -198
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/test_cov_corrwith.py +0 -181
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/test_dot_frame.py +0 -103
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/test_dot_series.py +0 -141
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/test_index.py +0 -109
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/test_series.py +0 -136
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/test_setitem_frame.py +0 -125
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/diff_frames_ops/test_setitem_series.py +0 -217
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/test_attrs.py +0 -384
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/test_constructor.py +0 -598
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/test_conversion.py +0 -73
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/test_reindexing.py +0 -869
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/test_reshaping.py +0 -487
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/test_spark.py +0 -309
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/test_take.py +0 -156
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/test_time_series.py +0 -149
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/frame/test_truncate.py +0 -163
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_aggregate.py +0 -311
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_apply_func.py +0 -524
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_cumulative.py +0 -419
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_describe.py +0 -144
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_groupby.py +0 -979
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_head_tail.py +0 -234
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_index.py +0 -206
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_missing_data.py +0 -421
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_split_apply.py +0 -187
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/groupby/test_stat.py +0 -397
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/test_align.py +0 -100
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/test_base.py +0 -2743
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/test_category.py +0 -484
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/test_datetime.py +0 -276
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/test_indexing.py +0 -432
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/test_reindex.py +0 -310
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/test_rename.py +0 -257
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/test_reset_index.py +0 -160
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/indexes/test_timedelta.py +0 -128
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/io/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/io/test_io.py +0 -137
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/plot/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/plot/test_frame_plot.py +0 -170
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/plot/test_frame_plot_matplotlib.py +0 -547
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/plot/test_frame_plot_plotly.py +0 -285
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/plot/test_series_plot.py +0 -106
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/plot/test_series_plot_matplotlib.py +0 -409
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/plot/test_series_plot_plotly.py +0 -247
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_all_any.py +0 -105
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_arg_ops.py +0 -197
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_as_of.py +0 -137
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_as_type.py +0 -227
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_compute.py +0 -634
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_conversion.py +0 -88
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_cumulative.py +0 -139
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_index.py +0 -475
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_missing_data.py +0 -265
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_series.py +0 -818
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_sort.py +0 -162
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/series/test_stat.py +0 -780
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_categorical.py +0 -741
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_config.py +0 -160
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_csv.py +0 -453
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_dataframe_conversion.py +0 -281
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_dataframe_spark_io.py +0 -487
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_default_index.py +0 -109
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_ewm.py +0 -434
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_expanding.py +0 -253
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_extension.py +0 -152
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_frame_spark.py +0 -162
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_generic_functions.py +0 -234
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_indexing.py +0 -1339
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_indexops_spark.py +0 -82
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_internal.py +0 -124
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_namespace.py +0 -638
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_numpy_compat.py +0 -200
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_ops_on_diff_frames.py +0 -1355
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_ops_on_diff_frames_groupby.py +0 -655
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_ops_on_diff_frames_groupby_expanding.py +0 -113
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_ops_on_diff_frames_groupby_rolling.py +0 -118
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_repr.py +0 -192
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_resample.py +0 -346
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_reshape.py +0 -495
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_rolling.py +0 -263
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_scalars.py +0 -59
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_series_conversion.py +0 -85
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_series_datetime.py +0 -364
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_series_string.py +0 -362
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_spark_functions.py +0 -46
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_sql.py +0 -123
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_stats.py +0 -581
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_typedef.py +0 -447
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_utils.py +0 -301
- snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/test_window.py +0 -465
- snowflake/snowpark_connect/includes/python/pyspark/resource/tests/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/resource/tests/test_resources.py +0 -83
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/client/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/client/test_artifact.py +0 -420
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/client/test_client.py +0 -358
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/streaming/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/streaming/test_parity_foreach.py +0 -36
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/streaming/test_parity_foreach_batch.py +0 -44
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/streaming/test_parity_listener.py +0 -116
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/streaming/test_parity_streaming.py +0 -35
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_connect_basic.py +0 -3612
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_connect_column.py +0 -1042
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_connect_function.py +0 -2381
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_connect_plan.py +0 -1060
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_arrow.py +0 -163
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_arrow_map.py +0 -38
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_arrow_python_udf.py +0 -48
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_catalog.py +0 -36
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_column.py +0 -55
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_conf.py +0 -36
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_dataframe.py +0 -96
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_datasources.py +0 -44
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_errors.py +0 -36
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_functions.py +0 -59
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_group.py +0 -36
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_pandas_cogrouped_map.py +0 -59
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_pandas_grouped_map.py +0 -74
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_pandas_grouped_map_with_state.py +0 -62
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_pandas_map.py +0 -58
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_pandas_udf.py +0 -70
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_pandas_udf_grouped_agg.py +0 -50
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_pandas_udf_scalar.py +0 -68
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_pandas_udf_window.py +0 -40
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_readwriter.py +0 -46
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_serde.py +0 -44
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_types.py +0 -100
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_udf.py +0 -100
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_parity_udtf.py +0 -163
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_session.py +0 -181
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/connect/test_utils.py +0 -42
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_cogrouped_map.py +0 -623
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_grouped_map.py +0 -869
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_grouped_map_with_state.py +0 -342
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_map.py +0 -436
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_udf.py +0 -363
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_udf_grouped_agg.py +0 -592
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_udf_scalar.py +0 -1503
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_udf_typehints.py +0 -392
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_udf_typehints_with_future_annotations.py +0 -375
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/pandas/test_pandas_udf_window.py +0 -411
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/streaming/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/streaming/test_streaming.py +0 -401
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/streaming/test_streaming_foreach.py +0 -295
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/streaming/test_streaming_foreach_batch.py +0 -106
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/streaming/test_streaming_listener.py +0 -558
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_arrow.py +0 -1346
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_arrow_map.py +0 -182
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_arrow_python_udf.py +0 -202
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_catalog.py +0 -503
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_column.py +0 -225
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_conf.py +0 -83
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_context.py +0 -201
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_dataframe.py +0 -1931
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_datasources.py +0 -256
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_errors.py +0 -69
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_functions.py +0 -1349
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_group.py +0 -53
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_pandas_sqlmetrics.py +0 -68
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_readwriter.py +0 -283
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_serde.py +0 -155
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_session.py +0 -412
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_types.py +0 -1581
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_udf.py +0 -961
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_udf_profiler.py +0 -165
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_udtf.py +0 -1456
- snowflake/snowpark_connect/includes/python/pyspark/sql/tests/test_utils.py +0 -1686
- snowflake/snowpark_connect/includes/python/pyspark/streaming/tests/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/streaming/tests/test_context.py +0 -184
- snowflake/snowpark_connect/includes/python/pyspark/streaming/tests/test_dstream.py +0 -706
- snowflake/snowpark_connect/includes/python/pyspark/streaming/tests/test_kinesis.py +0 -118
- snowflake/snowpark_connect/includes/python/pyspark/streaming/tests/test_listener.py +0 -160
- snowflake/snowpark_connect/includes/python/pyspark/tests/__init__.py +0 -16
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_appsubmit.py +0 -306
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_broadcast.py +0 -196
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_conf.py +0 -44
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_context.py +0 -346
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_daemon.py +0 -89
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_install_spark.py +0 -124
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_join.py +0 -69
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_memory_profiler.py +0 -167
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_pin_thread.py +0 -194
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_profiler.py +0 -168
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_rdd.py +0 -939
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_rddbarrier.py +0 -52
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_rddsampler.py +0 -66
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_readwrite.py +0 -368
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_serializers.py +0 -257
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_shuffle.py +0 -267
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_stage_sched.py +0 -153
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_statcounter.py +0 -130
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_taskcontext.py +0 -350
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_util.py +0 -97
- snowflake/snowpark_connect/includes/python/pyspark/tests/test_worker.py +0 -271
- snowpark_connect-0.23.0.dist-info/RECORD +0 -893
- {snowpark_connect-0.23.0.data → snowpark_connect-0.25.0.data}/scripts/snowpark-connect +0 -0
- {snowpark_connect-0.23.0.data → snowpark_connect-0.25.0.data}/scripts/snowpark-session +0 -0
- {snowpark_connect-0.23.0.data → snowpark_connect-0.25.0.data}/scripts/snowpark-submit +0 -0
- {snowpark_connect-0.23.0.dist-info → snowpark_connect-0.25.0.dist-info}/WHEEL +0 -0
- {snowpark_connect-0.23.0.dist-info → snowpark_connect-0.25.0.dist-info}/licenses/LICENSE-binary +0 -0
- {snowpark_connect-0.23.0.dist-info → snowpark_connect-0.25.0.dist-info}/licenses/LICENSE.txt +0 -0
- {snowpark_connect-0.23.0.dist-info → snowpark_connect-0.25.0.dist-info}/licenses/NOTICE-binary +0 -0
- {snowpark_connect-0.23.0.dist-info → snowpark_connect-0.25.0.dist-info}/top_level.txt +0 -0
|
@@ -1,309 +0,0 @@
|
|
|
1
|
-
#
|
|
2
|
-
# Licensed to the Apache Software Foundation (ASF) under one or more
|
|
3
|
-
# contributor license agreements. See the NOTICE file distributed with
|
|
4
|
-
# this work for additional information regarding copyright ownership.
|
|
5
|
-
# The ASF licenses this file to You under the Apache License, Version 2.0
|
|
6
|
-
# (the "License"); you may not use this file except in compliance with
|
|
7
|
-
# the License. You may obtain a copy of the License at
|
|
8
|
-
#
|
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
10
|
-
#
|
|
11
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
12
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
13
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
14
|
-
# See the License for the specific language governing permissions and
|
|
15
|
-
# limitations under the License.
|
|
16
|
-
#
|
|
17
|
-
import inspect
|
|
18
|
-
import sys
|
|
19
|
-
import unittest
|
|
20
|
-
from io import StringIO
|
|
21
|
-
|
|
22
|
-
import numpy as np
|
|
23
|
-
import pandas as pd
|
|
24
|
-
from pyspark import StorageLevel
|
|
25
|
-
from pyspark.ml.linalg import SparseVector
|
|
26
|
-
from pyspark.sql.types import StructType
|
|
27
|
-
|
|
28
|
-
from pyspark import pandas as ps
|
|
29
|
-
from pyspark.pandas.frame import CachedDataFrame
|
|
30
|
-
from pyspark.pandas.exceptions import PandasNotImplementedError
|
|
31
|
-
from pyspark.pandas.missing.frame import MissingPandasLikeDataFrame
|
|
32
|
-
from pyspark.testing.pandasutils import ComparisonTestBase, SPARK_CONF_ARROW_ENABLED
|
|
33
|
-
from pyspark.testing.sqlutils import SQLTestUtils
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
# This file contains test cases for 'Spark-related'
|
|
37
|
-
# https://spark.apache.org/docs/latest/api/python/reference/pyspark.pandas/frame.html#spark-related
|
|
38
|
-
class FrameSparkMixin:
|
|
39
|
-
@property
|
|
40
|
-
def pdf(self):
|
|
41
|
-
return pd.DataFrame(
|
|
42
|
-
{"a": [1, 2, 3, 4, 5, 6, 7, 8, 9], "b": [4, 5, 6, 3, 2, 1, 0, 0, 0]},
|
|
43
|
-
index=np.random.rand(9),
|
|
44
|
-
)
|
|
45
|
-
|
|
46
|
-
@property
|
|
47
|
-
def df_pair(self):
|
|
48
|
-
pdf = self.pdf
|
|
49
|
-
psdf = ps.from_pandas(pdf)
|
|
50
|
-
return pdf, psdf
|
|
51
|
-
|
|
52
|
-
def test_empty_dataframe(self):
|
|
53
|
-
pdf = pd.DataFrame({"a": pd.Series([], dtype="i1"), "b": pd.Series([], dtype="str")})
|
|
54
|
-
|
|
55
|
-
psdf = ps.from_pandas(pdf)
|
|
56
|
-
self.assert_eq(psdf, pdf)
|
|
57
|
-
|
|
58
|
-
with self.sql_conf({SPARK_CONF_ARROW_ENABLED: False}):
|
|
59
|
-
psdf = ps.from_pandas(pdf)
|
|
60
|
-
self.assert_eq(psdf, pdf)
|
|
61
|
-
|
|
62
|
-
def test_all_null_dataframe(self):
|
|
63
|
-
pdf = pd.DataFrame(
|
|
64
|
-
{
|
|
65
|
-
"a": [None, None, None, "a"],
|
|
66
|
-
"b": [None, None, None, 1],
|
|
67
|
-
"c": [None, None, None] + list(np.arange(1, 2).astype("i1")),
|
|
68
|
-
"d": [None, None, None, 1.0],
|
|
69
|
-
"e": [None, None, None, True],
|
|
70
|
-
"f": [None, None, None] + list(pd.date_range("20130101", periods=1)),
|
|
71
|
-
},
|
|
72
|
-
)
|
|
73
|
-
psdf = ps.from_pandas(pdf)
|
|
74
|
-
|
|
75
|
-
self.assert_eq(psdf.iloc[:-1], pdf.iloc[:-1])
|
|
76
|
-
|
|
77
|
-
with self.sql_conf({SPARK_CONF_ARROW_ENABLED: False}):
|
|
78
|
-
self.assert_eq(psdf.iloc[:-1], pdf.iloc[:-1])
|
|
79
|
-
|
|
80
|
-
pdf = pd.DataFrame(
|
|
81
|
-
{
|
|
82
|
-
"a": pd.Series([None, None, None], dtype="float64"),
|
|
83
|
-
"b": pd.Series([None, None, None], dtype="str"),
|
|
84
|
-
},
|
|
85
|
-
)
|
|
86
|
-
|
|
87
|
-
psdf = ps.from_pandas(pdf)
|
|
88
|
-
self.assert_eq(psdf, pdf)
|
|
89
|
-
|
|
90
|
-
with self.sql_conf({SPARK_CONF_ARROW_ENABLED: False}):
|
|
91
|
-
psdf = ps.from_pandas(pdf)
|
|
92
|
-
self.assert_eq(psdf, pdf)
|
|
93
|
-
|
|
94
|
-
def test_nullable_object(self):
|
|
95
|
-
pdf = pd.DataFrame(
|
|
96
|
-
{
|
|
97
|
-
"a": list("abc") + [np.nan, None],
|
|
98
|
-
"b": list(range(1, 4)) + [np.nan, None],
|
|
99
|
-
"c": list(np.arange(3, 6).astype("i1")) + [np.nan, None],
|
|
100
|
-
"d": list(np.arange(4.0, 7.0, dtype="float64")) + [np.nan, None],
|
|
101
|
-
"e": [True, False, True, np.nan, None],
|
|
102
|
-
"f": list(pd.date_range("20130101", periods=3)) + [np.nan, None],
|
|
103
|
-
},
|
|
104
|
-
index=np.random.rand(5),
|
|
105
|
-
)
|
|
106
|
-
|
|
107
|
-
psdf = ps.from_pandas(pdf)
|
|
108
|
-
self.assert_eq(psdf, pdf)
|
|
109
|
-
|
|
110
|
-
with self.sql_conf({SPARK_CONF_ARROW_ENABLED: False}):
|
|
111
|
-
psdf = ps.from_pandas(pdf)
|
|
112
|
-
self.assert_eq(psdf, pdf)
|
|
113
|
-
|
|
114
|
-
def test_dot_in_column_name(self):
|
|
115
|
-
self.assert_eq(
|
|
116
|
-
ps.DataFrame(ps.range(1)._internal.spark_frame.selectExpr("1L as `a.b`"))["a.b"],
|
|
117
|
-
ps.Series([1], name="a.b"),
|
|
118
|
-
)
|
|
119
|
-
|
|
120
|
-
def test_spark_schema(self):
|
|
121
|
-
psdf = ps.DataFrame(
|
|
122
|
-
{
|
|
123
|
-
"a": list("abc"),
|
|
124
|
-
"b": list(range(1, 4)),
|
|
125
|
-
"c": np.arange(3, 6).astype("i1"),
|
|
126
|
-
"d": np.arange(4.0, 7.0, dtype="float64"),
|
|
127
|
-
"e": [True, False, True],
|
|
128
|
-
"f": pd.date_range("20130101", periods=3),
|
|
129
|
-
},
|
|
130
|
-
columns=["a", "b", "c", "d", "e", "f"],
|
|
131
|
-
)
|
|
132
|
-
|
|
133
|
-
actual = psdf.spark.schema()
|
|
134
|
-
expected = (
|
|
135
|
-
StructType()
|
|
136
|
-
.add("a", "string", False)
|
|
137
|
-
.add("b", "long", False)
|
|
138
|
-
.add("c", "byte", False)
|
|
139
|
-
.add("d", "double", False)
|
|
140
|
-
.add("e", "boolean", False)
|
|
141
|
-
.add("f", "timestamp", False)
|
|
142
|
-
)
|
|
143
|
-
self.assertEqual(actual, expected)
|
|
144
|
-
|
|
145
|
-
actual = psdf.spark.schema("index")
|
|
146
|
-
expected = (
|
|
147
|
-
StructType()
|
|
148
|
-
.add("index", "long", False)
|
|
149
|
-
.add("a", "string", False)
|
|
150
|
-
.add("b", "long", False)
|
|
151
|
-
.add("c", "byte", False)
|
|
152
|
-
.add("d", "double", False)
|
|
153
|
-
.add("e", "boolean", False)
|
|
154
|
-
.add("f", "timestamp", False)
|
|
155
|
-
)
|
|
156
|
-
self.assertEqual(actual, expected)
|
|
157
|
-
|
|
158
|
-
def test_print_schema(self):
|
|
159
|
-
psdf = ps.DataFrame(
|
|
160
|
-
{"a": list("abc"), "b": list(range(1, 4)), "c": np.arange(3, 6).astype("i1")},
|
|
161
|
-
columns=["a", "b", "c"],
|
|
162
|
-
)
|
|
163
|
-
|
|
164
|
-
prev = sys.stdout
|
|
165
|
-
try:
|
|
166
|
-
out = StringIO()
|
|
167
|
-
sys.stdout = out
|
|
168
|
-
psdf.spark.print_schema()
|
|
169
|
-
actual = out.getvalue().strip()
|
|
170
|
-
|
|
171
|
-
self.assertTrue("a: string" in actual, actual)
|
|
172
|
-
self.assertTrue("b: long" in actual, actual)
|
|
173
|
-
self.assertTrue("c: byte" in actual, actual)
|
|
174
|
-
|
|
175
|
-
out = StringIO()
|
|
176
|
-
sys.stdout = out
|
|
177
|
-
psdf.spark.print_schema(index_col="index")
|
|
178
|
-
actual = out.getvalue().strip()
|
|
179
|
-
|
|
180
|
-
self.assertTrue("index: long" in actual, actual)
|
|
181
|
-
self.assertTrue("a: string" in actual, actual)
|
|
182
|
-
self.assertTrue("b: long" in actual, actual)
|
|
183
|
-
self.assertTrue("c: byte" in actual, actual)
|
|
184
|
-
finally:
|
|
185
|
-
sys.stdout = prev
|
|
186
|
-
|
|
187
|
-
def test_explain_hint(self):
|
|
188
|
-
psdf1 = ps.DataFrame(
|
|
189
|
-
{"lkey": ["foo", "bar", "baz", "foo"], "value": [1, 2, 3, 5]},
|
|
190
|
-
columns=["lkey", "value"],
|
|
191
|
-
)
|
|
192
|
-
psdf2 = ps.DataFrame(
|
|
193
|
-
{"rkey": ["foo", "bar", "baz", "foo"], "value": [5, 6, 7, 8]},
|
|
194
|
-
columns=["rkey", "value"],
|
|
195
|
-
)
|
|
196
|
-
merged = psdf1.merge(psdf2.spark.hint("broadcast"), left_on="lkey", right_on="rkey")
|
|
197
|
-
prev = sys.stdout
|
|
198
|
-
try:
|
|
199
|
-
out = StringIO()
|
|
200
|
-
sys.stdout = out
|
|
201
|
-
merged.spark.explain()
|
|
202
|
-
actual = out.getvalue().strip()
|
|
203
|
-
|
|
204
|
-
self.assertTrue("Broadcast" in actual, actual)
|
|
205
|
-
finally:
|
|
206
|
-
sys.stdout = prev
|
|
207
|
-
|
|
208
|
-
def test_cache(self):
|
|
209
|
-
pdf = pd.DataFrame(
|
|
210
|
-
[(0.2, 0.3), (0.0, 0.6), (0.6, 0.0), (0.2, 0.1)], columns=["dogs", "cats"]
|
|
211
|
-
)
|
|
212
|
-
psdf = ps.from_pandas(pdf)
|
|
213
|
-
|
|
214
|
-
with psdf.spark.cache() as cached_df:
|
|
215
|
-
self.assert_eq(isinstance(cached_df, CachedDataFrame), True)
|
|
216
|
-
self.assert_eq(
|
|
217
|
-
repr(cached_df.spark.storage_level), repr(StorageLevel(True, True, False, True))
|
|
218
|
-
)
|
|
219
|
-
|
|
220
|
-
def test_persist(self):
|
|
221
|
-
pdf = pd.DataFrame(
|
|
222
|
-
[(0.2, 0.3), (0.0, 0.6), (0.6, 0.0), (0.2, 0.1)], columns=["dogs", "cats"]
|
|
223
|
-
)
|
|
224
|
-
psdf = ps.from_pandas(pdf)
|
|
225
|
-
storage_levels = [
|
|
226
|
-
StorageLevel.DISK_ONLY,
|
|
227
|
-
StorageLevel.MEMORY_AND_DISK,
|
|
228
|
-
StorageLevel.MEMORY_ONLY,
|
|
229
|
-
StorageLevel.OFF_HEAP,
|
|
230
|
-
]
|
|
231
|
-
|
|
232
|
-
for storage_level in storage_levels:
|
|
233
|
-
with psdf.spark.persist(storage_level) as cached_df:
|
|
234
|
-
self.assert_eq(isinstance(cached_df, CachedDataFrame), True)
|
|
235
|
-
self.assert_eq(repr(cached_df.spark.storage_level), repr(storage_level))
|
|
236
|
-
|
|
237
|
-
self.assertRaises(TypeError, lambda: psdf.spark.persist("DISK_ONLY"))
|
|
238
|
-
|
|
239
|
-
def test_udt(self):
|
|
240
|
-
sparse_values = {0: 0.1, 1: 1.1}
|
|
241
|
-
sparse_vector = SparseVector(len(sparse_values), sparse_values)
|
|
242
|
-
pdf = pd.DataFrame({"a": [sparse_vector], "b": [10]})
|
|
243
|
-
|
|
244
|
-
psdf = ps.from_pandas(pdf)
|
|
245
|
-
self.assert_eq(psdf, pdf)
|
|
246
|
-
|
|
247
|
-
def test_missing(self):
|
|
248
|
-
psdf = self.psdf
|
|
249
|
-
|
|
250
|
-
missing_functions = inspect.getmembers(MissingPandasLikeDataFrame, inspect.isfunction)
|
|
251
|
-
unsupported_functions = [
|
|
252
|
-
name for (name, type_) in missing_functions if type_.__name__ == "unsupported_function"
|
|
253
|
-
]
|
|
254
|
-
for name in unsupported_functions:
|
|
255
|
-
with self.assertRaisesRegex(
|
|
256
|
-
PandasNotImplementedError,
|
|
257
|
-
"method.*DataFrame.*{}.*not implemented( yet\\.|\\. .+)".format(name),
|
|
258
|
-
):
|
|
259
|
-
getattr(psdf, name)()
|
|
260
|
-
|
|
261
|
-
deprecated_functions = [
|
|
262
|
-
name for (name, type_) in missing_functions if type_.__name__ == "deprecated_function"
|
|
263
|
-
]
|
|
264
|
-
for name in deprecated_functions:
|
|
265
|
-
with self.assertRaisesRegex(
|
|
266
|
-
PandasNotImplementedError, "method.*DataFrame.*{}.*is deprecated".format(name)
|
|
267
|
-
):
|
|
268
|
-
getattr(psdf, name)()
|
|
269
|
-
|
|
270
|
-
missing_properties = inspect.getmembers(
|
|
271
|
-
MissingPandasLikeDataFrame, lambda o: isinstance(o, property)
|
|
272
|
-
)
|
|
273
|
-
unsupported_properties = [
|
|
274
|
-
name
|
|
275
|
-
for (name, type_) in missing_properties
|
|
276
|
-
if type_.fget.__name__ == "unsupported_property"
|
|
277
|
-
]
|
|
278
|
-
for name in unsupported_properties:
|
|
279
|
-
with self.assertRaisesRegex(
|
|
280
|
-
PandasNotImplementedError,
|
|
281
|
-
"property.*DataFrame.*{}.*not implemented( yet\\.|\\. .+)".format(name),
|
|
282
|
-
):
|
|
283
|
-
getattr(psdf, name)
|
|
284
|
-
deprecated_properties = [
|
|
285
|
-
name
|
|
286
|
-
for (name, type_) in missing_properties
|
|
287
|
-
if type_.fget.__name__ == "deprecated_property"
|
|
288
|
-
]
|
|
289
|
-
for name in deprecated_properties:
|
|
290
|
-
with self.assertRaisesRegex(
|
|
291
|
-
PandasNotImplementedError, "property.*DataFrame.*{}.*is deprecated".format(name)
|
|
292
|
-
):
|
|
293
|
-
getattr(psdf, name)
|
|
294
|
-
|
|
295
|
-
|
|
296
|
-
class FrameSparkTests(FrameSparkMixin, ComparisonTestBase, SQLTestUtils):
|
|
297
|
-
pass
|
|
298
|
-
|
|
299
|
-
|
|
300
|
-
if __name__ == "__main__":
|
|
301
|
-
from pyspark.pandas.tests.frame.test_spark import * # noqa: F401
|
|
302
|
-
|
|
303
|
-
try:
|
|
304
|
-
import xmlrunner
|
|
305
|
-
|
|
306
|
-
testRunner = xmlrunner.XMLTestRunner(output="target/test-reports", verbosity=2)
|
|
307
|
-
except ImportError:
|
|
308
|
-
testRunner = None
|
|
309
|
-
unittest.main(testRunner=testRunner, verbosity=2)
|
|
@@ -1,156 +0,0 @@
|
|
|
1
|
-
#
|
|
2
|
-
# Licensed to the Apache Software Foundation (ASF) under one or more
|
|
3
|
-
# contributor license agreements. See the NOTICE file distributed with
|
|
4
|
-
# this work for additional information regarding copyright ownership.
|
|
5
|
-
# The ASF licenses this file to You under the Apache License, Version 2.0
|
|
6
|
-
# (the "License"); you may not use this file except in compliance with
|
|
7
|
-
# the License. You may obtain a copy of the License at
|
|
8
|
-
#
|
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
10
|
-
#
|
|
11
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
12
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
13
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
14
|
-
# See the License for the specific language governing permissions and
|
|
15
|
-
# limitations under the License.
|
|
16
|
-
#
|
|
17
|
-
import unittest
|
|
18
|
-
|
|
19
|
-
import numpy as np
|
|
20
|
-
import pandas as pd
|
|
21
|
-
|
|
22
|
-
from pyspark import pandas as ps
|
|
23
|
-
from pyspark.testing.pandasutils import ComparisonTestBase
|
|
24
|
-
from pyspark.testing.sqlutils import SQLTestUtils
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
class FrameTakeMixin:
|
|
28
|
-
@property
|
|
29
|
-
def pdf(self):
|
|
30
|
-
return pd.DataFrame(
|
|
31
|
-
{"a": [1, 2, 3, 4, 5, 6, 7, 8, 9], "b": [4, 5, 6, 3, 2, 1, 0, 0, 0]},
|
|
32
|
-
index=np.random.rand(9),
|
|
33
|
-
)
|
|
34
|
-
|
|
35
|
-
@property
|
|
36
|
-
def df_pair(self):
|
|
37
|
-
pdf = self.pdf
|
|
38
|
-
psdf = ps.from_pandas(pdf)
|
|
39
|
-
return pdf, psdf
|
|
40
|
-
|
|
41
|
-
def test_take(self):
|
|
42
|
-
pdf = pd.DataFrame(
|
|
43
|
-
{"A": range(0, 50000), "B": range(100000, 0, -2), "C": range(100000, 50000, -1)}
|
|
44
|
-
)
|
|
45
|
-
psdf = ps.from_pandas(pdf)
|
|
46
|
-
|
|
47
|
-
# axis=0 (default)
|
|
48
|
-
self.assert_eq(psdf.take([1, 2]).sort_index(), pdf.take([1, 2]).sort_index())
|
|
49
|
-
self.assert_eq(psdf.take([-1, -2]).sort_index(), pdf.take([-1, -2]).sort_index())
|
|
50
|
-
self.assert_eq(
|
|
51
|
-
psdf.take(range(100, 110)).sort_index(), pdf.take(range(100, 110)).sort_index()
|
|
52
|
-
)
|
|
53
|
-
self.assert_eq(
|
|
54
|
-
psdf.take(range(-110, -100)).sort_index(), pdf.take(range(-110, -100)).sort_index()
|
|
55
|
-
)
|
|
56
|
-
self.assert_eq(
|
|
57
|
-
psdf.take([10, 100, 1000, 10000]).sort_index(),
|
|
58
|
-
pdf.take([10, 100, 1000, 10000]).sort_index(),
|
|
59
|
-
)
|
|
60
|
-
self.assert_eq(
|
|
61
|
-
psdf.take([-10, -100, -1000, -10000]).sort_index(),
|
|
62
|
-
pdf.take([-10, -100, -1000, -10000]).sort_index(),
|
|
63
|
-
)
|
|
64
|
-
|
|
65
|
-
# axis=1
|
|
66
|
-
self.assert_eq(
|
|
67
|
-
psdf.take([1, 2], axis=1).sort_index(), pdf.take([1, 2], axis=1).sort_index()
|
|
68
|
-
)
|
|
69
|
-
self.assert_eq(
|
|
70
|
-
psdf.take([-1, -2], axis=1).sort_index(), pdf.take([-1, -2], axis=1).sort_index()
|
|
71
|
-
)
|
|
72
|
-
self.assert_eq(
|
|
73
|
-
psdf.take(range(1, 3), axis=1).sort_index(),
|
|
74
|
-
pdf.take(range(1, 3), axis=1).sort_index(),
|
|
75
|
-
)
|
|
76
|
-
self.assert_eq(
|
|
77
|
-
psdf.take(range(-1, -3), axis=1).sort_index(),
|
|
78
|
-
pdf.take(range(-1, -3), axis=1).sort_index(),
|
|
79
|
-
)
|
|
80
|
-
self.assert_eq(
|
|
81
|
-
psdf.take([2, 1], axis=1).sort_index(),
|
|
82
|
-
pdf.take([2, 1], axis=1).sort_index(),
|
|
83
|
-
)
|
|
84
|
-
self.assert_eq(
|
|
85
|
-
psdf.take([-1, -2], axis=1).sort_index(),
|
|
86
|
-
pdf.take([-1, -2], axis=1).sort_index(),
|
|
87
|
-
)
|
|
88
|
-
|
|
89
|
-
# MultiIndex columns
|
|
90
|
-
columns = pd.MultiIndex.from_tuples([("A", "Z"), ("B", "X"), ("C", "C")])
|
|
91
|
-
psdf.columns = columns
|
|
92
|
-
pdf.columns = columns
|
|
93
|
-
|
|
94
|
-
# MultiIndex columns with axis=0 (default)
|
|
95
|
-
self.assert_eq(psdf.take([1, 2]).sort_index(), pdf.take([1, 2]).sort_index())
|
|
96
|
-
self.assert_eq(psdf.take([-1, -2]).sort_index(), pdf.take([-1, -2]).sort_index())
|
|
97
|
-
self.assert_eq(
|
|
98
|
-
psdf.take(range(100, 110)).sort_index(), pdf.take(range(100, 110)).sort_index()
|
|
99
|
-
)
|
|
100
|
-
self.assert_eq(
|
|
101
|
-
psdf.take(range(-110, -100)).sort_index(), pdf.take(range(-110, -100)).sort_index()
|
|
102
|
-
)
|
|
103
|
-
self.assert_eq(
|
|
104
|
-
psdf.take([10, 100, 1000, 10000]).sort_index(),
|
|
105
|
-
pdf.take([10, 100, 1000, 10000]).sort_index(),
|
|
106
|
-
)
|
|
107
|
-
self.assert_eq(
|
|
108
|
-
psdf.take([-10, -100, -1000, -10000]).sort_index(),
|
|
109
|
-
pdf.take([-10, -100, -1000, -10000]).sort_index(),
|
|
110
|
-
)
|
|
111
|
-
|
|
112
|
-
# axis=1
|
|
113
|
-
self.assert_eq(
|
|
114
|
-
psdf.take([1, 2], axis=1).sort_index(), pdf.take([1, 2], axis=1).sort_index()
|
|
115
|
-
)
|
|
116
|
-
self.assert_eq(
|
|
117
|
-
psdf.take([-1, -2], axis=1).sort_index(), pdf.take([-1, -2], axis=1).sort_index()
|
|
118
|
-
)
|
|
119
|
-
self.assert_eq(
|
|
120
|
-
psdf.take(range(1, 3), axis=1).sort_index(),
|
|
121
|
-
pdf.take(range(1, 3), axis=1).sort_index(),
|
|
122
|
-
)
|
|
123
|
-
self.assert_eq(
|
|
124
|
-
psdf.take(range(-1, -3), axis=1).sort_index(),
|
|
125
|
-
pdf.take(range(-1, -3), axis=1).sort_index(),
|
|
126
|
-
)
|
|
127
|
-
self.assert_eq(
|
|
128
|
-
psdf.take([2, 1], axis=1).sort_index(),
|
|
129
|
-
pdf.take([2, 1], axis=1).sort_index(),
|
|
130
|
-
)
|
|
131
|
-
self.assert_eq(
|
|
132
|
-
psdf.take([-1, -2], axis=1).sort_index(),
|
|
133
|
-
pdf.take([-1, -2], axis=1).sort_index(),
|
|
134
|
-
)
|
|
135
|
-
|
|
136
|
-
# Checking the type of indices.
|
|
137
|
-
self.assertRaises(TypeError, lambda: psdf.take(1))
|
|
138
|
-
self.assertRaises(TypeError, lambda: psdf.take("1"))
|
|
139
|
-
self.assertRaises(TypeError, lambda: psdf.take({1, 2}))
|
|
140
|
-
self.assertRaises(TypeError, lambda: psdf.take({1: None, 2: None}))
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
class FrameTakeTests(FrameTakeMixin, ComparisonTestBase, SQLTestUtils):
|
|
144
|
-
pass
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
if __name__ == "__main__":
|
|
148
|
-
from pyspark.pandas.tests.frame.test_take import * # noqa: F401
|
|
149
|
-
|
|
150
|
-
try:
|
|
151
|
-
import xmlrunner
|
|
152
|
-
|
|
153
|
-
testRunner = xmlrunner.XMLTestRunner(output="target/test-reports", verbosity=2)
|
|
154
|
-
except ImportError:
|
|
155
|
-
testRunner = None
|
|
156
|
-
unittest.main(testRunner=testRunner, verbosity=2)
|
|
@@ -1,149 +0,0 @@
|
|
|
1
|
-
#
|
|
2
|
-
# Licensed to the Apache Software Foundation (ASF) under one or more
|
|
3
|
-
# contributor license agreements. See the NOTICE file distributed with
|
|
4
|
-
# this work for additional information regarding copyright ownership.
|
|
5
|
-
# The ASF licenses this file to You under the Apache License, Version 2.0
|
|
6
|
-
# (the "License"); you may not use this file except in compliance with
|
|
7
|
-
# the License. You may obtain a copy of the License at
|
|
8
|
-
#
|
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
10
|
-
#
|
|
11
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
12
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
13
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
14
|
-
# See the License for the specific language governing permissions and
|
|
15
|
-
# limitations under the License.
|
|
16
|
-
#
|
|
17
|
-
from datetime import datetime
|
|
18
|
-
import unittest
|
|
19
|
-
|
|
20
|
-
import numpy as np
|
|
21
|
-
import pandas as pd
|
|
22
|
-
|
|
23
|
-
from pyspark import pandas as ps
|
|
24
|
-
from pyspark.testing.pandasutils import ComparisonTestBase
|
|
25
|
-
from pyspark.testing.sqlutils import SQLTestUtils
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
# This file contains test cases for 'Time series-related'
|
|
29
|
-
# https://spark.apache.org/docs/latest/api/python/reference/pyspark.pandas/frame.html#time-series-related
|
|
30
|
-
class FrameTimeSeriesMixin:
|
|
31
|
-
@property
|
|
32
|
-
def pdf(self):
|
|
33
|
-
return pd.DataFrame(
|
|
34
|
-
{"a": [1, 2, 3, 4, 5, 6, 7, 8, 9], "b": [4, 5, 6, 3, 2, 1, 0, 0, 0]},
|
|
35
|
-
index=np.random.rand(9),
|
|
36
|
-
)
|
|
37
|
-
|
|
38
|
-
@property
|
|
39
|
-
def df_pair(self):
|
|
40
|
-
pdf = self.pdf
|
|
41
|
-
psdf = ps.from_pandas(pdf)
|
|
42
|
-
return pdf, psdf
|
|
43
|
-
|
|
44
|
-
def test_shift(self):
|
|
45
|
-
pdf = pd.DataFrame(
|
|
46
|
-
{
|
|
47
|
-
"Col1": [10, 20, 15, 30, 45],
|
|
48
|
-
"Col2": [13, 23, 18, 33, 48],
|
|
49
|
-
"Col3": [17, 27, 22, 37, 52],
|
|
50
|
-
},
|
|
51
|
-
index=np.random.rand(5),
|
|
52
|
-
)
|
|
53
|
-
psdf = ps.from_pandas(pdf)
|
|
54
|
-
|
|
55
|
-
self.assert_eq(pdf.shift(3), psdf.shift(3))
|
|
56
|
-
self.assert_eq(pdf.shift().shift(-1), psdf.shift().shift(-1))
|
|
57
|
-
self.assert_eq(pdf.shift().sum().astype(int), psdf.shift().sum())
|
|
58
|
-
|
|
59
|
-
# Need the expected result since pandas 0.23 does not support `fill_value` argument.
|
|
60
|
-
pdf1 = pd.DataFrame(
|
|
61
|
-
{"Col1": [0, 0, 0, 10, 20], "Col2": [0, 0, 0, 13, 23], "Col3": [0, 0, 0, 17, 27]},
|
|
62
|
-
index=pdf.index,
|
|
63
|
-
)
|
|
64
|
-
self.assert_eq(pdf1, psdf.shift(periods=3, fill_value=0))
|
|
65
|
-
msg = "should be an int"
|
|
66
|
-
with self.assertRaisesRegex(TypeError, msg):
|
|
67
|
-
psdf.shift(1.5)
|
|
68
|
-
|
|
69
|
-
# multi-index columns
|
|
70
|
-
columns = pd.MultiIndex.from_tuples([("x", "Col1"), ("x", "Col2"), ("y", "Col3")])
|
|
71
|
-
pdf.columns = columns
|
|
72
|
-
psdf.columns = columns
|
|
73
|
-
self.assert_eq(pdf.shift(3), psdf.shift(3))
|
|
74
|
-
self.assert_eq(pdf.shift().shift(-1), psdf.shift().shift(-1))
|
|
75
|
-
self.assert_eq(pdf.shift(0), psdf.shift(0))
|
|
76
|
-
|
|
77
|
-
def test_first_valid_index(self):
|
|
78
|
-
pdf = pd.DataFrame(
|
|
79
|
-
{"a": [None, 2, 3, 2], "b": [None, 2.0, 3.0, 1.0], "c": [None, 200, 400, 200]},
|
|
80
|
-
index=["Q", "W", "E", "R"],
|
|
81
|
-
)
|
|
82
|
-
psdf = ps.from_pandas(pdf)
|
|
83
|
-
self.assert_eq(pdf.first_valid_index(), psdf.first_valid_index())
|
|
84
|
-
self.assert_eq(pdf[[]].first_valid_index(), psdf[[]].first_valid_index())
|
|
85
|
-
|
|
86
|
-
# MultiIndex columns
|
|
87
|
-
pdf.columns = pd.MultiIndex.from_tuples([("a", "x"), ("b", "y"), ("c", "z")])
|
|
88
|
-
psdf = ps.from_pandas(pdf)
|
|
89
|
-
self.assert_eq(pdf.first_valid_index(), psdf.first_valid_index())
|
|
90
|
-
|
|
91
|
-
# Empty DataFrame
|
|
92
|
-
pdf = pd.Series([]).to_frame()
|
|
93
|
-
psdf = ps.Series([]).to_frame()
|
|
94
|
-
self.assert_eq(pdf.first_valid_index(), psdf.first_valid_index())
|
|
95
|
-
|
|
96
|
-
pdf = pd.DataFrame(
|
|
97
|
-
{"a": [None, 2, 3, 2], "b": [None, 2.0, 3.0, 1.0], "c": [None, 200, 400, 200]},
|
|
98
|
-
index=[
|
|
99
|
-
datetime(2021, 1, 1),
|
|
100
|
-
datetime(2021, 2, 1),
|
|
101
|
-
datetime(2021, 3, 1),
|
|
102
|
-
datetime(2021, 4, 1),
|
|
103
|
-
],
|
|
104
|
-
)
|
|
105
|
-
psdf = ps.from_pandas(pdf)
|
|
106
|
-
self.assert_eq(pdf.first_valid_index(), psdf.first_valid_index())
|
|
107
|
-
|
|
108
|
-
def test_last_valid_index(self):
|
|
109
|
-
pdf = pd.DataFrame(
|
|
110
|
-
{"a": [1, 2, 3, None], "b": [1.0, 2.0, 3.0, None], "c": [100, 200, 400, None]},
|
|
111
|
-
index=["Q", "W", "E", "R"],
|
|
112
|
-
)
|
|
113
|
-
psdf = ps.from_pandas(pdf)
|
|
114
|
-
self.assert_eq(pdf.last_valid_index(), psdf.last_valid_index())
|
|
115
|
-
self.assert_eq(pdf[[]].last_valid_index(), psdf[[]].last_valid_index())
|
|
116
|
-
|
|
117
|
-
# MultiIndex columns
|
|
118
|
-
pdf.columns = pd.MultiIndex.from_tuples([("a", "x"), ("b", "y"), ("c", "z")])
|
|
119
|
-
psdf = ps.from_pandas(pdf)
|
|
120
|
-
self.assert_eq(pdf.last_valid_index(), psdf.last_valid_index())
|
|
121
|
-
|
|
122
|
-
# Empty DataFrame
|
|
123
|
-
pdf = pd.Series([]).to_frame()
|
|
124
|
-
psdf = ps.Series([]).to_frame()
|
|
125
|
-
self.assert_eq(pdf.last_valid_index(), psdf.last_valid_index())
|
|
126
|
-
|
|
127
|
-
def test_to_datetime(self):
|
|
128
|
-
pdf = pd.DataFrame(
|
|
129
|
-
{"year": [2015, 2016], "month": [2, 3], "day": [4, 5]}, index=np.random.rand(2)
|
|
130
|
-
)
|
|
131
|
-
psdf = ps.from_pandas(pdf)
|
|
132
|
-
|
|
133
|
-
self.assert_eq(pd.to_datetime(pdf), ps.to_datetime(psdf))
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
class FrameTimeSeriesTests(FrameTimeSeriesMixin, ComparisonTestBase, SQLTestUtils):
|
|
137
|
-
pass
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
if __name__ == "__main__":
|
|
141
|
-
from pyspark.pandas.tests.frame.test_time_series import * # noqa: F401
|
|
142
|
-
|
|
143
|
-
try:
|
|
144
|
-
import xmlrunner
|
|
145
|
-
|
|
146
|
-
testRunner = xmlrunner.XMLTestRunner(output="target/test-reports", verbosity=2)
|
|
147
|
-
except ImportError:
|
|
148
|
-
testRunner = None
|
|
149
|
-
unittest.main(testRunner=testRunner, verbosity=2)
|