mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
Replace assert with unittest assertions (#13957)
This commit is contained in:
parent
6e4c8f683c
commit
e1bb2ebd92
1 changed files with 3 additions and 3 deletions
|
|
@ -27,6 +27,6 @@ class CLITest(unittest.TestCase):
|
|||
|
||||
with CaptureStd() as cs:
|
||||
transformers.commands.transformers_cli.main()
|
||||
assert "Python version" in cs.out
|
||||
assert "Platform" in cs.out
|
||||
assert "Using distributed or parallel set-up in script?" in cs.out
|
||||
self.assertIn("Python version", cs.out)
|
||||
self.assertIn("Platform", cs.out)
|
||||
self.assertIn("Using distributed or parallel set-up in script?", cs.out)
|
||||
|
|
|
|||
Loading…
Reference in a new issue