1. Lighthouse

Level 1 · Check the split before the result

The request

From Jordan Reyes, Head of Marketing

“The free-shipping banner test ran all of November and December. Before you tell me whether it worked, our data lead says you should check the split first. Can you confirm the two groups are the size they should be?”

Your goal

Using test (sessions from 2025-11-01 to 2025-12-31, with arm = "A" or "B"), set result to a dict with "A" and "B", the number of sessions in each arm (int), "chi2", the chi-square statistic against a 50/50 split (rounded to 3), and "p_value", its p-value with 1 degree of freedom, 2 * (1 - NormalDist().cdf(sqrt(chi2))) (rounded to 4).

  • Use the web_sessions DataFramenot done yet
  • Store a dict in resultnot done yet

Your code

Python

Loading the editor…

    ResultPreview of web_sessions · press Run to see yours
    Preview of web_sessions
    session_idcustomer_idstarted_atdevicesource
    1NaN2023-01-01 23:02:24mobilepaid
    2182023-01-01 15:07:49desktoppaid
    3NaN2023-01-01 16:55:46desktopdirect
    4NaN2023-01-01 12:43:28mobileorganic

    Mote checks your result

    Press Run. Each row that matches the goal becomes a firefly in your lantern.

    DataFrames

    Loaded for you as pandas DataFrames, plus pd and np.

    web_sessions40,000 rows · 7 columns
    • session_idinteger
    • customer_idinteger
    • started_attimestamp
    • devicevarchar
    • sourcevarchar
    • pages_viewedinteger
    • convertedboolean
    First 5 rows of web_sessions
    session_idcustomer_idstarted_atdevicesourcepages_viewedconverted
    1NaN2023-01-01 23:02:24mobilepaid1false
    2182023-01-01 15:07:49desktoppaid1false
    3NaN2023-01-01 16:55:46desktopdirect7false
    4NaN2023-01-01 12:43:28mobileorganic3false
    5NaN2023-01-01 13:06:21mobilesocial2false
    customers2,018 rows · 9 columns
    • customer_idinteger
    • first_namevarchar
    • last_namevarchar
    • emailvarchar
    • cityvarchar
    • statevarchar
    • signup_datedate
    • segmentvarchar
    • referral_sourcevarchar
    First 5 rows of customers
    customer_idfirst_namelast_nameemailcitystatesignup_datesegmentreferral_source
    1OwenHudsonowen.hudson@icloud.comSan FranciscoCA2022-06-02consumersearch
    2DouglasMendozadmendoza69@gmail.comAtlantaGA2022-06-02consumerNaN
    3HenryBrookshbrooks@lakesidecafe.comLos AngelesCA2022-06-04businessNaN
    4BrianGrahambgraham@bluecedarhomes.comSeattleWA2022-06-06businesspinterest
    5KyleJimenezkjimenez27@yahoo.comIndianapolisIN2022-06-07consumerNaN
    products120 rows · 8 columns
    • product_idinteger
    • namevarchar
    • categoryvarchar
    • subcategoryvarchar
    • unit_pricedecimal
    • unit_costdecimal
    • launched_ondate
    • is_activeboolean
    First 5 rows of products
    product_idnamecategorysubcategoryunit_priceunit_costlaunched_onis_active
    101Vale Velvet LoveseatFurnitureSofas1659.00NaN2023-09-11true
    102Alder Oak Coffee TableFurnitureTables919.00503.502024-05-16true
    103Orchard Rattan Dining ChairFurnitureChairs789.00429.872023-05-27true
    104Upland Oak BookcaseFurnitureStorage429.00214.702022-12-20true
    105Avery Leather Bed FrameFurnitureBeds449.00232.702021-08-31true
    orders12,000 rows · 7 columns
    • order_idinteger
    • customer_idinteger
    • order_datetimestamp
    • statusvarchar
    • channelvarchar
    • shipping_feedecimal
    • coupon_codevarchar
    First 5 rows of orders
    order_idcustomer_idorder_datestatuschannelshipping_feecoupon_code
    10011252023-01-01 20:51:19completedweb0.00NaN
    10021562023-01-02 07:58:46completedweb0.00NaN
    1003222023-01-02 10:25:27completedmarketplace4.99WELCOME10
    1004482023-01-02 11:21:51completedmarketplace4.99WELCOME10
    1005822023-01-02 11:43:36completedweb0.00WELCOME10
    order_items30,338 rows · 6 columns
    • order_item_idinteger
    • order_idinteger
    • product_idinteger
    • quantityinteger
    • unit_pricedecimal
    • discount_pctdecimal
    First 5 rows of order_items
    order_item_idorder_idproduct_idquantityunit_pricediscount_pct
    11001121931.990.00
    21001154311.990.00
    31002154411.990.00
    410021941149.000.00
    51002193219.990.00
    returns1,673 rows · 5 columns
    • return_idinteger
    • order_item_idinteger
    • returned_ondate
    • reasonvarchar
    • refund_amountdecimal
    First 5 rows of returns
    return_idorder_item_idreturned_onreasonrefund_amount
    1142023-01-10damaged83.69
    2422023-01-10wrong_item19.99
    31052023-01-14wrong_item10.99
    4242023-01-16changed_mind26.99
    51262023-01-16late20.99
    employees60 rows · 8 columns
    • employee_idinteger
    • first_namevarchar
    • last_namevarchar
    • departmentvarchar
    • titlevarchar
    • manager_idinteger
    • hire_datedate
    • salaryinteger
    First 5 rows of employees
    employee_idfirst_namelast_namedepartmenttitlemanager_idhire_datesalary
    1CatherineHaleExecutiveChief Executive OfficerNaN2016-03-01285000
    2RobertKimExecutiveChief Operating Officer12017-01-09235000
    3MeganCollinsSalesHead of Sales12018-04-16172000
    4ChloeGutierrezSalesAccount Executive32020-01-2172000
    5FrankBellSalesAccount Executive32023-05-2173000
    marketing_spend3,276 rows · 3 columns
    • spend_datedate
    • channelvarchar
    • amountdecimal
    First 5 rows of marketing_spend
    spend_datechannelamount
    2023-01-01paid492.99
    2023-01-01email37.38
    2023-01-01social200.53
    2023-01-02paid709.13
    2023-01-02email64.13

    The lesson

    Why check the split first

    An A/B test promises one thing: the two groups differ only in what you changed. If the assignment is broken, that promise is gone and the result means nothing, however clean it looks. So before you look at conversion, you check that each arm got the share of traffic it was supposed to get. That check has a name: the sample ratio mismatch test, or SRM.

    Harbor & Pine’s banner test split sessions by session_id: even ids saw the banner (arm B), odd ids did not (arm A). The plan was 50/50, so each arm should hold half of the sessions in the window.

    test["arm"].value_counts()
    # A    2604
    # B    2604

    The chi-square check

    “Close to half” is not a test. The chi-square goodness-of-fit statistic measures how far the counts are from what the plan predicts:

    expected = len(test) / 2
    chi2 = (n["A"] - expected) ** 2 / expected + (n["B"] - expected) ** 2 / expected

    With two arms there is one degree of freedom, and a chi-square with one degree of freedom is the square of a standard normal variable. So you can get its p-value from the normal distribution in Python’s standard library, no SciPy needed:

    from statistics import NormalDist
    p = 2 * (1 - NormalDist().cdf(math.sqrt(chi2)))

    Here the arms are exactly equal, so chi2 is 0 and the p-value is 1. The split is healthy.

    What a failing check looks like

    Suppose arm B had lost 3% of its sessions to a slow banner script: 2,604 in A and 2,526 in B. Then chi2 is about 1.19 and p is about 0.28, which still passes. Scale the same 3% loss up to a test of 50,000 sessions per arm and p drops below 0.0001. SRM checks get stricter as tests get bigger, which is exactly why big companies run them automatically. Most teams treat a p-value under 0.001 as a failed split and stop the analysis.

    Common mistakes

    • Eyeballing the counts. 50.6% versus 49.4% looks fine and can be a real bug on a large test.
    • Cutting the window wrong. started_at < "2025-12-31" drops the whole last day, because the timestamps on December 31 are later than midnight. Use < "2026-01-01" for an inclusive end date.
    • Checking only after the result disappoints. Run the SRM check first, every time, whatever the result says. A check you only run on bad news is a way of choosing the answer you like.

    Interview note

    “What would you check before trusting an A/B test result?” The sample ratio is the first answer interviewers hope to hear, followed by the test window, the unit of randomisation and the metric definition.

    Pick what may run on this site. You can change this any time from "Cookie settings" at the bottom of every page.