Skip to content

Commit

Permalink
Fix delete function
Browse files Browse the repository at this point in the history
  • Loading branch information
argenisleon committed Dec 13, 2019
1 parent cd00cbe commit 6248fcf
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion optimus/helpers/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -269,9 +269,10 @@ def add_buffer(value):

# return "".join(buffer)

def delete(self, func, suffix=None):
def delete(self, df, func, suffix=None, *args, **kwargs):
"""
This is a helper function that delete python tests files used to construct the final Test file.
:param df: Do nothing, only for simplicity so you can delete a file test the same way you create it
:param suffix: The create method will try to create a test function with the func param given.
If you want to test a function with different params you can use suffix.
:param func: Spark dataframe function to be tested
Expand Down

0 comments on commit 6248fcf

Please sign in to comment.