diff --git a/pieces/ProphetPredictPiece/test_example_simple_piece.py b/pieces/ProphetPredictPiece/test_example_simple_piece.py deleted file mode 100644 index 168a611..0000000 --- a/pieces/ProphetPredictPiece/test_example_simple_piece.py +++ /dev/null @@ -1,15 +0,0 @@ -from domino.testing import piece_dry_run - -def test_example_simple_piece(): - input_data = dict( - distribution_name="gaussian", - distribution_mean=0., - distribution_sd=1. - ) - output_data = piece_dry_run( - "ExampleSimplePiece", - input_data - ) - - assert output_data["message"] is not None - assert output_data["sample_result"] is not None \ No newline at end of file diff --git a/pieces/ProphetPredictPiece/test_prophet_predict_piece.py b/pieces/ProphetPredictPiece/test_prophet_predict_piece.py new file mode 100644 index 0000000..af4c3dd --- /dev/null +++ b/pieces/ProphetPredictPiece/test_prophet_predict_piece.py @@ -0,0 +1,4 @@ +from domino.testing import piece_dry_run + +def test_prophet_predict_piece(): + ... \ No newline at end of file diff --git a/pieces/ProphetTrainModelPiece/test_example_simple_piece.py b/pieces/ProphetTrainModelPiece/test_example_simple_piece.py deleted file mode 100644 index 168a611..0000000 --- a/pieces/ProphetTrainModelPiece/test_example_simple_piece.py +++ /dev/null @@ -1,15 +0,0 @@ -from domino.testing import piece_dry_run - -def test_example_simple_piece(): - input_data = dict( - distribution_name="gaussian", - distribution_mean=0., - distribution_sd=1. - ) - output_data = piece_dry_run( - "ExampleSimplePiece", - input_data - ) - - assert output_data["message"] is not None - assert output_data["sample_result"] is not None \ No newline at end of file diff --git a/pieces/ProphetTrainModelPiece/test_prophet_train_model_piece.py b/pieces/ProphetTrainModelPiece/test_prophet_train_model_piece.py new file mode 100644 index 0000000..bf0946c --- /dev/null +++ b/pieces/ProphetTrainModelPiece/test_prophet_train_model_piece.py @@ -0,0 +1,4 @@ +from domino.testing import piece_dry_run + +def test_prophet_train_model_piece(): + ... \ No newline at end of file