NHSDigital
diff --git a/‎codonPython/check_null.py‎
Lines changed: 2 additions & 2 deletions b/‎codonPython/check_null.py‎
Lines changed: 2 additions & 2 deletions
@@ -1,7 +1,7 @@
 import numpy
 import pandas as pd
 
-def check_null(dataframe: pd.DataFrame, columns_to_be_checked: list) -> bool:
+def check_null(dataframe: pd.DataFrame, columns_to_be_checked: list) -> int:
     """
     Checks a pandas dataframe for null values
 
@@ -23,7 +23,7 @@ def check_null(dataframe: pd.DataFrame, columns_to_be_checked: list) -> bool:
     --------
     >>> check_null(dataframe = pd.DataFrame({'col1': [1,2], 'col2': [3,4]}),columns_to_be_checked = ['col1', 'col2'])
     0
-    >>> check_null(dataframe = pd.DataFrame({'col1': [1,numpy.nan], 'col2': [3,4]}),columns_to_be_checked = ['col1'])
+    >>> check_null(dataframe = pd.DataFrame({'col1': [numpy.nan,numpy.nan], 'col2': [3,4]}),columns_to_be_checked = ['col1'])
     1
     """
Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`import numpy`
`2`	`2`	`import pandas as pd`
`3`	`3`
`4`		`-def check_null(dataframe: pd.DataFrame, columns_to_be_checked: list) -> bool:`
	`4`	`+def check_null(dataframe: pd.DataFrame, columns_to_be_checked: list) -> int:`
`5`	`5`	`"""`
`6`	`6`	`Checks a pandas dataframe for null values`
`7`	`7`
`@@ -23,7 +23,7 @@ def check_null(dataframe: pd.DataFrame, columns_to_be_checked: list) -> bool:`
`23`	`23`	`--------`
`24`	`24`	`>>> check_null(dataframe = pd.DataFrame({'col1': [1,2], 'col2': [3,4]}),columns_to_be_checked = ['col1', 'col2'])`
`25`	`25`	`0`
`26`		`- >>> check_null(dataframe = pd.DataFrame({'col1': [1,numpy.nan], 'col2': [3,4]}),columns_to_be_checked = ['col1'])`
	`26`	`+ >>> check_null(dataframe = pd.DataFrame({'col1': [numpy.nan,numpy.nan], 'col2': [3,4]}),columns_to_be_checked = ['col1'])`
`27`	`27`	`1`
`28`	`28`	`"""`
`29`	`29`