from rembed.markdown.extension import REmbedExtension
import markdown

from hamcrest import assert_that, contains_string


def test_should_get_correct_embedding():
    md = markdown.Markdown(extensions=[REmbedExtension()])
    embedding = md.convert(
        '[!rembed](https://twitter.com/BarackObama/status/266031293945503744)')
    assert_that(embedding, contains_string('Four more years.'))

@@ -17,26 +17,16 @@
 down
-.extension
  import 
 REmb
@@ -21,23 +21,17 @@
  import 
-REmbedE
+e
 xtension
@@ -180,16 +180,26 @@
 nsions=%5B
+extension.
 REmbedEx

Refactor import