Ejemplo n.º 1
0
    def test_load_features_pass_trajectory_as_topology(self):
        import mdtraj
        t = mdtraj.load(pdb_file)
        features = api.featurizer(t)

        traj_files = [f for f in self.bpti_mini_files if f.endswith('.netcdf')]
        load(traj_files, features=features)
Ejemplo n.º 2
0
    def test_load_trajs(self):
        features = api.featurizer(pdb_file)
        res = load(traj_files, features)

        self.assertEqual(type(res), list)
        self.assertTrue(all(type(x) == np.ndarray for x in res))
Ejemplo n.º 3
0
    def test_load_traj(self):
        filename = traj_files[0]
        features = api.featurizer(pdb_file)
        res = load(filename, features)

        self.assertEqual(type(res), np.ndarray)
Ejemplo n.º 4
0
    def testUnicodeString(self):
        filename = text_type(traj_files[0])
        features = api.featurizer(pdb_file)

        load(filename, features)
Ejemplo n.º 5
0
    def testUnicodeString(self):
        filename = u''.join(traj_files[0])
        features = api.featurizer(pdb_file)

        load(filename, features)
Ejemplo n.º 6
0
    def test_load_trajs(self):
        features = api.featurizer(pdb_file)
        res = load(traj_files, features)

        self.assertEqual(type(res), list)
        self.assertTrue(all(type(x) == np.ndarray for x in res))
Ejemplo n.º 7
0
    def test_load_traj(self):
        filename = traj_files[0]
        features = api.featurizer(pdb_file)
        res = load(filename, features)

        self.assertEqual(type(res), np.ndarray)
Ejemplo n.º 8
0
    def testUnicodeString(self):
        filename = unicode(traj_files[0])
        features = api.featurizer(pdb_file)

        load(filename, features)