import docopt
from script import Script
from scripts import suite

doc = suite.__doc__


class TestSuite(Script):
    script_name = 'teuthology-suite'

    def test_repo(self):
        assert suite.expand_short_repo_name('foo', 'https://github.com/ceph/ceph-ci.git') == 'https://github.com/ceph/foo'
        assert suite.expand_short_repo_name('foo/bar', 'https://github.com/ceph/ceph-ci.git') == 'https://github.com/foo/bar'
    
    def test_args(self):
        args = docopt.docopt(doc, [
            "--ceph-repo", "foo",
            "--suite-repo", "foo/bar"
        ])
        conf = suite.process_args(args)
        assert args["ceph_repo"] == "https://github.com/ceph/foo"
        assert args["suite_repo"] == "https://github.com/foo/bar"

@@ -1,18 +1,4 @@
-import docopt%0A
 from
@@ -23,55 +23,8 @@
 ipt%0A
-from scripts import suite%0A%0Adoc = suite.__doc__%0A
 %0A%0Acl
@@ -87,600 +87,4 @@
 te'%0A
-%0A    def test_repo(self):%0A        assert suite.expand_short_repo_name('foo', 'https://github.com/ceph/ceph-ci.git') == 'https://github.com/ceph/foo'%0A        assert suite.expand_short_repo_name('foo/bar', 'https://github.com/ceph/ceph-ci.git') == 'https://github.com/foo/bar'%0A    %0A    def test_args(self):%0A        args = docopt.docopt(doc, %5B%0A            %22--ceph-repo%22, %22foo%22,%0A            %22--suite-repo%22, %22foo/bar%22%0A        %5D)%0A        conf = suite.process_args(args)%0A        assert args%5B%22ceph_repo%22%5D == %22https://github.com/ceph/foo%22%0A        assert args%5B%22suite_repo%22%5D == %22https://github.com/foo/bar%22%0A

Revert changes to scripts/test/test_suite.py