def test_it_returns_the_right_columns(self): df = build_dataframe_from_users(self.users) self.assertCountEqual(df.columns, [ "bsas_ocurrencias", "bsas_usuarios", "cba_ocurrencias", "cba_usuarios", ])
def test_it_counts_in_cba(self): df = build_dataframe_from_users(self.users) self.assertEqual(df.loc["hola"]["cba_ocurrencias"], 1) self.assertEqual(df.loc["mundo"]["cba_ocurrencias"], 1) self.assertEqual(df.loc["culia"]["cba_ocurrencias"], 3) self.assertEqual(df.loc["gato"]["cba_ocurrencias"], 5) self.assertEqual(df.loc["hola"]["cba_usuarios"], 1) self.assertEqual(df.loc["mundo"]["cba_usuarios"], 1) self.assertEqual(df.loc["culia"]["cba_usuarios"], 3) self.assertEqual(df.loc["gato"]["cba_usuarios"], 2)
def test_it_counts_users(self): df = build_dataframe_from_users(self.users) self.assertEqual(df.loc["hola"]["bsas_usuarios"], 1) self.assertEqual(df.loc["mundo"]["bsas_usuarios"], 1)
def test_it_returns_the_given_words(self): df = build_dataframe_from_users(self.users) self.assertCountEqual(df.index, ["hola", "mundo", "gato", "culia"])
def test_it_counts_occurrences(self): df = build_dataframe_from_users(self.users) self.assertEqual(df.loc["hola"]["bsas_ocurrencias"], 2) self.assertEqual(df.loc["mundo"]["bsas_ocurrencias"], 2)