Commit e8144401 authored by nd-02110114's avatar nd-02110114
Browse files

🐛 fix ci error

parent a7e173f7
Loading
Loading
Loading
Loading
+4 −3
Original line number Diff line number Diff line
@@ -2,6 +2,7 @@
Tests for Pose Generation
"""
import os
import platform
import tempfile
import unittest
import logging
@@ -9,18 +10,18 @@ import numpy as np
import deepchem as dc
import pytest

IS_WINDOWS = platform.system() == 'Windows'

class TestPoseGeneration(unittest.TestCase):
  """
  Does sanity checks on pose generation.
  """

  @pytest.mark.linux_only
  @unittest.skipIf(IS_WINDOWS, 'Skip the test on Windows')
  def test_vina_initialization(self):
    """Test that VinaPoseGenerator can be initialized."""
    dc.dock.VinaPoseGenerator()

  @pytest.mark.linux_only
  @unittest.skipIf(IS_WINDOWS, 'Skip the test on Windows')
  def test_pocket_vina_initialization(self):
    """Test that VinaPoseGenerator can be initialized."""
    pocket_finder = dc.dock.ConvexHullPocketFinder()