def test_extract(self): expected = "some text" assert expected == extract("123 some text 456", "123", "456") expected = "123 some text" assert expected == extract("123 some text 456", None, "456") expected = "some text 456" assert expected == extract("123 some text 456", "123", None) expected = "value" assert expected == extract("test(value) test2(value2)", "test(", ")") expected = "value" assert expected == extract("test:value", ":") expected = """Task ----""" string = """ some text Task ---- some more text""" result = extract(string, "some text", "some more text") assert result == expected
def test_extract_error_02(self): try: extract(None, "sub1", "sub2") assert False except: assert True
def test_extract_error_01(self): try: extract("some string", None, None) assert False except: assert True