def testEq(self):
copied = WorkspaceGen3(self._testWorkspace)
self.assertEqual(self._testbed, copied)
with tempfile.TemporaryDirectory() as temp:
different = WorkspaceGen3(temp)
self.assertNotEqual(self._testbed, different)
self.assertNotEqual(copied, different)
def setUp(self):
self._testDir = tempfile.mkdtemp()
self.addCleanup(shutil.rmtree, self._testDir, ignore_errors=True)
# Fake Butler to avoid Workspace initialization overhead
self.setUpMockPatch("lsst.daf.butler.Registry",
**{"queryDatasets.return_value": []})
self.setUpMockPatch("lsst.daf.butler.Butler")
self.workspace = WorkspaceGen3(self._testDir)
self.apPipeArgs = pipeline_driver.ApPipeParser().parse_args(
["--id", "visit = %d" % _getDataIds()[0]["visit"]])
def testMakeDir(self):
"""Verify that a Workspace creates the workspace directory if it does not exist.
"""
newPath = '_temp3' # can't use mkdtemp because creation is what we're testing
shutil.rmtree(newPath, ignore_errors=True)
self.assertFalse(os.path.exists(newPath),
'Workspace directory must not exist before test.')
try:
WorkspaceGen3(newPath)
self.assertTrue(os.path.exists(newPath),
'Workspace directory must exist.')
finally:
shutil.rmtree(newPath, ignore_errors=True)
def setUp(self):
super().setUp()
self.config = self.makeTestConfig()
self.config.validate()
self.config.freeze()
self.root = tempfile.mkdtemp()
self.addCleanup(shutil.rmtree, self.root, ignore_errors=True)
self.workspace = WorkspaceGen3(self.root)
self.task = ingestion.Gen3DatasetIngestTask(config=self.config,
dataset=self.dataset, workspace=self.workspace)
self.butler = self.workspace.workButler
def setUp(self):
# Use realpath to avoid link problems
self._testWorkspace = os.path.realpath(tempfile.mkdtemp())
self._testbed = WorkspaceGen3(self._testWorkspace)