Skip to main content

assert_valid_values

assert_valid_values()

Asserts that only the given valid values present in the given column. This assertion can be used with Layer dataset entities.

  • Parameters

    • column_name (str) -- Column name to be checked from the given dataframe.
    • valid_values (List[Any**]) -- Valid values.
  • Returns

    Function object.

  • Return type

    Callable[[...], Any]

from layer.decorators.assertions import assert_valid_values
from layer.decorators import dataset

@dataset("user-os-data")
@assert_valid_values("OperatingSystem", ["linux", "other"])
def create_my_dataset():
data = [["user1", "linux"], ["user2", "linux"], ["user3", "other"]]
dataframe = pd.DataFrame(data, columns=["UserName", "OperatingSystem"])
return dataframe