def test_same_fit_transform(self): Y, Y_rdd = self.make_dense_randint_rdd(low=0, high=10, shape=(1000,)) local = LabelEncoder() dist = SparkLabelEncoder() assert_array_equal(local.fit_transform(Y), dist.fit_transform(Y_rdd).toarray())
def test_same_fit_transform(self): Y, Y_rdd = self.make_dense_randint_rdd(low=0, high=10, shape=(1000, )) local = LabelEncoder() dist = SparkLabelEncoder() assert_array_equal(local.fit_transform(Y), dist.fit_transform(Y_rdd).toarray())