Unify separation tools and add exclusive control mode
This commit is contained in:
1 parent
088f5f6f91
commit
bd5aea7656
13 files changed
+81
-28
No files matched your search
@@ -54,4 +54,11 @@ class Tests(unittest.TestCase):
|
||||
with tempfile.TemporaryDirectory() as root:
|
||||
worker=SeparatorTests(root,Mock(),Mock());directory=Path(root,'a'*32);directory.mkdir();(directory/'complete').touch();(directory/'results.json').write_text('["../../secret.wav"]')
|
||||
with self.assertRaises(ValueError):worker.audio('a'*32,0)
|
||||
|
||||
def test_full_job_requires_separator_mode_before_thread_starts(self):
|
||||
with tempfile.TemporaryDirectory() as root:
|
||||
scheduler=Mock();scheduler.gpu_mode='llm'
|
||||
worker=SeparatorTests(root,Mock(),scheduler)
|
||||
with self.assertRaisesRegex(ValueError,'Steuerung'):worker.start(next(iter(MODELS)),audio(),full=True)
|
||||
|
||||
if __name__=='__main__':unittest.main()
|
||||
Reference in new issue
Block a user