Skip to content

Commit

Permalink
updated tests with new imports
Browse files Browse the repository at this point in the history
  • Loading branch information
Christina Bukas committed Jul 12, 2023
1 parent 79fef41 commit 69c6720
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 7 deletions.
17 changes: 11 additions & 6 deletions src/client/test/test_app.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,9 @@
import unittest

from dcp_client.app import Application
from dcp_client.bentoml_model import BentomlModel
from dcp_client.fsimagestorage import FilesystemImageStorage

from dcp_client.utils.bentoml_model import BentomlModel
from dcp_client.utils.fsimagestorage import FilesystemImageStorage
from dcp_client.utils.sync_src_dst import DataRSync

class TestApplication(unittest.TestCase):

Expand All @@ -23,10 +23,15 @@ def test_load_image(self):
os.mkdir('in_prog')
imsave('in_prog/test_img.png', img)
imsave('in_prog/test_img2.png', img2)

rsyncer = DataRSync(user_name="local",
host_name="local",
server_repo_path='.')
self.app = Application(BentomlModel(),
FilesystemImageStorage(),
inprogr_data_path='in_prog')
rsyncer,
FilesystemImageStorage(),
"0.0.0.0",
7010)

self.app.cur_selected_img = 'test_img.png'
self.app.cur_selected_path = 'in_prog'

Expand Down
2 changes: 1 addition & 1 deletion src/client/test/test_fsimagestorage.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from skimage import data
import unittest

from dcp_client.fsimagestorage import FilesystemImageStorage
from dcp_client.utils.fsimagestorage import FilesystemImageStorage


class TestFilesystemImageStorage(unittest.TestCase):
Expand Down

0 comments on commit 69c6720

Please sign in to comment.