def test_ntimescales_1(): # see issue #603 trajs = [np.random.randint(0, 30, size=500) for _ in range(5)] pccap = PCCAPlus(n_macrostates=11).fit(trajs) lumped_trajs = pccap.transform(trajs) assert len(np.unique(lumped_trajs)) == 11
def test_ntimescales_1(): # see issue #603 trajs = [random.randint(0, 100, size=500) for _ in range(15)] pccap = PCCAPlus(n_macrostates=11).fit(trajs) lumped_trajs = pccap.transform(trajs) observed_macros = len(np.unique(lumped_trajs)) assert observed_macros == 11, observed_macros