def test_get_distinct_l_with_nulls(self): iris = pd.read_csv("tests/iris.csv") iris = iris.append(pd.DataFrame([[1, 1, None, 1, "Iris-Test"], [1, 1, None, 1, "Iris-Test"], [1, 1, 2, 1, "Iris-Test"]], columns=iris.columns.values)) l_diversity = anonymization.get_distinct_l(iris, groupby=['Name'], column='PetalLength') self.assertEqual(l_diversity, 3)
def test_get_distinct_l(self): iris = pd.read_csv("tests/iris.csv") l_diversity = anonymization.get_distinct_l(iris, groupby=['Name'], column='PetalLength') self.assertEqual(l_diversity, 9)