def _get_tc_counts(Nt1o1, Nt0o1, Nt1o0, Nt0o0):
    """Get treatment and control group sizes from `_get_counts` output.

    Parameters
    ----------
    Nt1o1 : int
        Number of entries where (treatment, outcome) == (1,1).
    Nt0o1 : int
        Number of entries where (treatment, outcome) == (0,1).
    Nt1o0 : int
        Number of entries where (treatment, outcome) == (1,0).
    Nt0o0 : int
        Number of entries where (treatment, outcome) == (0,0).

    Returns
    -------
    Nt1 : int
        Size of treatment group.
    Nt0 : int
        Size of control group.
    N : int
        Size of full group.

    """
    Nt1 = Nt1o0 + Nt1o1
    Nt0 = Nt0o0 + Nt0o1
    N = Nt0 + Nt1
    return Nt1, Nt0, N