X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=test%2Fhelper.py;fp=test%2Fhelper.py;h=651ef99b983973dab1f17ab8619849af2b9fe9b1;hb=23d9ded655f45c5a21c8d93ad167a072322ae88f;hp=805467c82b0eceab526decb4244793066a529977;hpb=4c1a017e6961c18fdef13d49bcdd0daa08ab0575;p=youtube-dl diff --git a/test/helper.py b/test/helper.py index 805467c82..651ef99b9 100644 --- a/test/helper.py +++ b/test/helper.py @@ -103,6 +103,16 @@ def expect_info_dict(self, got_dict, expected_dict): self.assertTrue( match_rex.match(got), 'field %s (value: %r) should match %r' % (info_field, got, match_str)) + elif isinstance(expected, compat_str) and expected.startswith('startswith:'): + got = got_dict.get(info_field) + start_str = expected[len('startswith:'):] + self.assertTrue( + isinstance(got, compat_str), + 'Expected a %s object, but got %s for field %s' % ( + compat_str.__name__, type(got).__name__, info_field)) + self.assertTrue( + got.startswith(start_str), + 'field %s (value: %r) should start with %r' % (info_field, got, start_str)) elif isinstance(expected, type): got = got_dict.get(info_field) self.assertTrue(isinstance(got, expected),