Since this is an example, pls dont use this unit-test like style, but rather either generate random data or even load a classification dataset (see other examples)