#43 add categorial distribution to the data generator integration test