import unittest
from fragalysis_api import GetPdbData
from fragalysis_api import GetMoleculesData

class GetPdbDataTest(unittest.TestCase):

    def test_GetPdbData_exists(self):
        search = GetPdbData()
        self.assertIsNotNone(search)

    def test_get_apo_pdb_file(self):
        search = GetPdbData()
        self.assertIsNotNone(search.get_apo_pdb_file('NUDT5A-x0114_1'))

if __name__ == '__main__':
    unittest.main()
