def test_some_results(get_dataframe): """ ContactBalance() returns a dataframe that contains hand-picked results. """ df = get_dataframe(ContactBalance("2016-01-01", "2016-01-07")) set_df = df.set_index("subscriber") assert set_df.loc["bvEWVnZdwJ8Lgkm2"]["proportion"] == 1.000000 assert set_df.loc["7XebRKr35JMJnq8A"]["events"] == 12 assert (set_df.loc["3XKdxqvyNxO2vLD1"]["msisdn_counterpart"].values[0] == "DELmRj9Vvl346G50") df = get_dataframe( ContactBalance("2016-01-01", "2016-01-07", direction="in")) set_df = df.set_index("subscriber") assert set_df.loc["bvEWVnZdwJ8Lgkm2"]["proportion"] == 1.000000 assert set_df.loc["8lo9EgjnyjgKO7vL"]["events"] == 19 assert set_df.loc["3XKdxqvyNxO2vLD1"][ "msisdn_counterpart"] == "7lNP0mDOAK3xKWv4" df = get_dataframe( ContactBalance("2016-01-01", "2016-01-07", direction="out")) set_df = df.set_index("subscriber") assert set_df.loc["V1QBpMj0vEwr2PGW"]["proportion"] == 1.000000 assert set_df.loc["7XebRKr35JMJnq8A"]["events"] == 12 assert set_df.loc["3XKdxqvyNxO2vLD1"][ "msisdn_counterpart"] == "DELmRj9Vvl346G50"
def test_no_result_is_greater_than_one(get_dataframe): """ No results from ContactBalance()['proportion'] is greater than 1. """ df = get_dataframe(ContactBalance("2016-01-01", "2016-01-07")) results = df[df["proportion"] > 1] assert len(results) == 0
def test_some_results(get_dataframe): """ TotalSubscriberEvents() returns a dataframe that contains hand-picked results. """ df = get_dataframe(ContactBalance("2016-01-01", "2016-01-07")) set_df = df.set_index("subscriber") assert set_df.loc["bvEWVnZdwJ8Lgkm2"]["proportion"] == 1.000000 assert set_df.loc["7XebRKr35JMJnq8A"]["events"] == 12 assert (set_df.loc["3XKdxqvyNxO2vLD1"]["msisdn_counterpart"].values[0] == "DELmRj9Vvl346G50")
def test_counterpart_subset(get_dataframe): """Test that counterparts_subset method gets correct subset.""" query = ContactBalance("2016-01-01", "2016-01-03") cb = get_dataframe(query) cs = get_dataframe(query.counterparts_subset()) assert set(cb.msisdn_counterpart.values) == set(cs.subscriber.values) query = ContactBalance("2016-01-01", "2016-01-03") cb = get_dataframe(query) cs = get_dataframe(query.counterparts_subset(include_subscribers=True)) assert set(cb.msisdn_counterpart.values).union(cb.subscriber.values) == set( cs.subscriber.values ) query = ContactBalance("2016-01-01", "2016-01-03", subscriber_identifier="imei") cb = get_dataframe(query) cs = get_dataframe(query.counterparts_subset()) assert set(cb.msisdn_counterpart.values) == set(cs.subscriber.values) query = ContactBalance("2016-01-01", "2016-01-03", subscriber_identifier="imei") cb = get_dataframe(query) with pytest.raises(ValueError): cs = get_dataframe(query.counterparts_subset(include_subscribers=True))