Skip to content

Commit

Permalink
Added scannet tests
Browse files Browse the repository at this point in the history
  • Loading branch information
AldovdN committed Jul 27, 2022
1 parent 5313c8d commit 0c50cc4
Show file tree
Hide file tree
Showing 5 changed files with 1,695 additions and 11 deletions.
17 changes: 11 additions & 6 deletions src/cport/modules/scannet.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,14 +129,19 @@ def parse_prediction(self, url=None, test_file=None):
"""
parser = PDB.PDBParser()
browser = ms.StatefulBrowser()
if not test_file:
browser = ms.StatefulBrowser()

browser.open(url)
pdb_string = re.findall(
r"stringContainingTheWholePdbFile = (.*?);", str(browser.page), re.DOTALL
)[0]

browser.open(url)
pdb_string = re.findall(
r"stringContainingTheWholePdbFile = (.*?);", str(browser.page), re.DOTALL
)[0]
structure = parser.get_structure("pdb", io.StringIO(pdb_string))

else:
structure = parser.get_structure("pdb", test_file)

structure = parser.get_structure("pdb", io.StringIO(pdb_string))
model = structure[0]
chain = model[self.chain_id]

Expand Down
Loading

0 comments on commit 0c50cc4

Please sign in to comment.