import pytest
import cleaner

class TestTagRemoval():
    def test_span_removal(self):
        text = ('<span style="font-family: &quot;helvetica neue&quot; ,'
        '&quot;arial&quot; , &quot;helvetica&quot; , sans-serif;">This is some'
        ' dummy text lalalala</span> This is some more dummy text '
        '<span>test</span>')

        expected = ('This is some dummy text lalalala This is some more dummy '
        'text test')

        cleaned = cleaner.remove_superflous_markup(text)

        assert cleaned == expected

import pytest
import cleaner

class TestTagTools():
    def test_get_pure_tag(self):
        tag1 = '<div>'
        tag2 = '</div>'
        tag3 = '<pre class="prettyprint">'

        assert cleaner.get_pure_tag(tag1) == '<div>'
        assert cleaner.get_pure_tag(tag2) == '</div>'
        assert cleaner.get_pure_tag(tag3) == '<pre>'

Add test for getting pure html tag
Add test for getting pure html tag

  import pytest
  import cleaner
  
- class TestTagRemoval():
+ class TestTagTools():
-     def test_span_removal(self):
+     def test_get_pure_tag(self):
+         tag1 = '<div>'
+         tag2 = '</div>'
+         tag3 = '<pre class="prettyprint">'
-         text = ('<span style="font-family: &quot;helvetica neue&quot; ,'
-         '&quot;arial&quot; , &quot;helvetica&quot; , sans-serif;">This is some'
-         ' dummy text lalalala</span> This is some more dummy text '
-         '<span>test</span>')
  
-         expected = ('This is some dummy text lalalala This is some more dummy '
-         'text test')
+         assert cleaner.get_pure_tag(tag1) == '<div>'
+         assert cleaner.get_pure_tag(tag2) == '</div>'
+         assert cleaner.get_pure_tag(tag3) == '<pre>'
  
-         cleaned = cleaner.remove_superflous_markup(text)
- 
-         assert cleaned == expected
- 
Add test for getting pure html tag
## Code Before:
import pytest
import cleaner

class TestTagRemoval():
    def test_span_removal(self):
        text = ('<span style="font-family: &quot;helvetica neue&quot; ,'
        '&quot;arial&quot; , &quot;helvetica&quot; , sans-serif;">This is some'
        ' dummy text lalalala</span> This is some more dummy text '
        '<span>test</span>')

        expected = ('This is some dummy text lalalala This is some more dummy '
        'text test')

        cleaned = cleaner.remove_superflous_markup(text)

        assert cleaned == expected

## Instruction:
Add test for getting pure html tag
## Code After:
import pytest
import cleaner

class TestTagTools():
    def test_get_pure_tag(self):
        tag1 = '<div>'
        tag2 = '</div>'
        tag3 = '<pre class="prettyprint">'

        assert cleaner.get_pure_tag(tag1) == '<div>'
        assert cleaner.get_pure_tag(tag2) == '</div>'
        assert cleaner.get_pure_tag(tag3) == '<pre>'
