test if correctly loaded

HouGroup · Dec 19, 2023 · 6afce60 · 6afce60
1 parent 467c099
commit 6afce60
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mdgo/forcefield/crawler.py b/mdgo/forcefield/crawler.py
@@ -110,6 +110,7 @@ def data_from_pdb(self, pdb_dir: str):
             pdb_dir: The path to the input pdb structure file.
         """
         self.web.get("http://traken.chem.yale.edu/ligpargen/")
+        self.wait.until(EC.presence_of_element_located((By.XPATH, "/html/body/div[2]/div/div[1]/ul/li[6]/h5")))
         upload_xpath = '//*[@id="exampleMOLFile"]'
         time.sleep(1)
         self.wait.until(EC.presence_of_element_located((By.XPATH, upload_xpath)))