def testBipartiteToUni(self):
userInds = numpy.array([0, 1, 2, 0, 4])
itemInds = numpy.array([0, 0, 0, 1, 1])
dates = numpy.array([1, 2, 3, 4, 5])
dataset = MendeleyGroupsDataset()
user1Inds, user2Inds, newDates = dataset.bipartiteToUni(userInds, itemInds, dates)
X = scipy.sparse.csc_matrix((newDates, (user1Inds, user2Inds)))
nptst.assert_array_equal(X[0, :].nonzero()[1], numpy.array([1, 2, 4]))
nptst.assert_array_equal(X[1, :].nonzero()[1], numpy.array([0, 2]))
nptst.assert_array_equal(X[2, :].nonzero()[1], numpy.array([0, 1]))
nptst.assert_array_equal(X[3, :].nonzero()[1], numpy.array([]))
nptst.assert_array_equal(X[4, :].nonzero()[1], numpy.array([0]))
