def test_get_total_by_cr_when_cr_is_negative_should_ignore(spark_session): sales_data = [(-1, 'mango', 0, 50, 50), (2, 'manzana', 2, 10, 10), (2, 'mango', 2, 40, 80)] sales_df = spark_session.createDataFrame(sales_data, get_schema()) expected = spark_session.createDataFrame([(2, 90)], ['numer_caja', 'ventas']) actual = get_total_by_cash_register(sales_df) assert expected.collect() == actual.collect()
def test_get_total_by_cr_when_only_one_sale(spark_session): sales_data = [(1, 'mango', 1, 50, 50)] sales_df = spark_session.createDataFrame(sales_data, get_schema()) expected = spark_session.createDataFrame([(1, 50)], ['numer_caja', 'ventas']) actual = get_total_by_cash_register(sales_df) assert expected.collect() == actual.collect()
def test_get_total_by_cr_happy_path(spark_session): sales_data = [(1, 'mango', 1, 50, 50), (2, 'manzana', 2, 10, 10), (2, 'mango', 2, 40, 80)] sales_df = spark_session.createDataFrame(sales_data, get_schema()) expected = spark_session.createDataFrame([(2, 90),(1, 50)], ['numer_caja', 'ventas']) actual = get_total_by_cash_register(sales_df) assert expected.collect() == actual.collect()
def test_get_total_by_cr_when_data_set_is_null(): sales_df = None actual = get_total_by_cash_register(sales_df) assert actual is None