Close #8: Properly rename classes with TestCase in their names

Author: dannysepler

.gitignore

@@ -1,8 +1,7 @@
 *.egg-info
 *.pyc
 /.coverage
+/.hypothesis
 /.mypy_cache
 /.pytest_cache
-/.tox
-/venv*
 dist

CHANGELOG.md

@@ -4,6 +4,11 @@ All notable changes to this project will be documented in this file. This projec
 
 ## [Unreleased]
 
+
+## [1.0.8] - September 19th 2021
+
+- Fix: A class called `ThingTestCase` should be renamed to `TestThing`
+
 ## [1.0.7] - September 16th 2021
 
 - Another edge case involving multi-line comments and trailing slashes

pytestify/fixes/base_class.py

@@ -40,6 +40,7 @@ def remove_base_class(contents: str) -> str:
         line = content_list[i]
         for variation in variations:
             line = line.replace(variation, '')
+            orig_name = orig_name.replace(variation, '')
 
         # delete empty paren if they exist
         line = line.replace(f'{orig_name}():', f'{orig_name}:')

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = pytestify
-version = 1.0.7
+version = 1.0.8
 description = Automatically convert unittests to pytest
 long_description = file: README.md
 long_description_content_type = text/markdown

tests/fixes/test_base_class.py

@@ -11,6 +11,11 @@
         ('class TestCls(TestCase): pass', 'class TestCls: pass'),
         ('class ClsTests(TestCase): pass', 'class TestCls: pass'),
         ('class ClsTest(TestCase): pass', 'class TestCls: pass'),
+        (
+            'class ThingTestCase(unittest.TestCase): pass',
+            'class TestThing: pass',
+        ),
+        ('class ThingTestCase(TestCase): pass', 'class TestThing: pass'),
     ],
 )
 def test_remove_base_class(before, after):