fixes

Author: kopekC

python2_to_python3/guide.md

@@ -2,7 +2,6 @@
 from codegen import Codebase
 
 # Initialize codebase
-codebase = Codebase("./")
 
 # Define the target directory
 TARGET_DIR = "input_repo"
@@ -151,4 +150,6 @@ def run():
 
 
 if __name__ == "__main__":
+    codebase = Codebase("./")
+
     run(codebase)

python2_to_python3/run.py

@@ -0,0 +1,28 @@
+# jj_classes/castle.py
+
+class Castle:
+    """Defines the Castle class."""
+
+    def __init__(self, name):
+        """Initialize the castle."""
+        if not name:
+            raise ValueError("Castle name cannot be empty.")
+        self._name = name
+        self._boss = "Bowser"
+        self._world = "Grass Land"
+
+    def has_access(self, character):
+        """Check if a character has access to the castle."""
+        return character.powerup == "Super Mushroom"
+
+    @property
+    def name(self):
+        return self._name
+
+    @property
+    def boss(self):
+        return self._boss
+
+    @property
+    def world(self):
+        return self._world

unittest_to_pytest/input_repo/jj_classes/__init__.py

@@ -0,0 +1,23 @@
+# jj_classes/character.py
+
+class Character:
+    """Defines the Character class."""
+
+    def __init__(self, name):
+        """Initialize the character."""
+        if not name:
+            raise ValueError("Character name cannot be empty.")
+        self._name = name
+        self._powerup = None
+
+    @property
+    def name(self):
+        return self._name
+
+    @property
+    def powerup(self):
+        return self._powerup
+
+    @powerup.setter
+    def powerup(self, value):
+        self._powerup = value

unittest_to_pytest/input_repo/jj_classes/castle.py

@@ -0,0 +1,9 @@
+# run_tests.py
+
+import unittest
+
+if __name__ == "__main__":
+    loader = unittest.TestLoader()
+    tests = loader.discover("tests")
+    test_runner = unittest.TextTestRunner()
+    test_runner.run(tests)

unittest_to_pytest/input_repo/jj_classes/character.py

@@ -0,0 +1,90 @@
+# tests/test_classes.py
+
+import unittest
+from unittest.mock import Mock
+from jj_classes.castle import Castle
+from jj_classes.character import Character
+
+
+class TestCastle(unittest.TestCase):
+    """Tests for the Castle class."""
+
+    def setUp(self):
+        """Set up a test castle."""
+        self.castle = Castle("Test Castle")
+
+    def test_castle_name(self):
+        """Test that the castle name is set correctly."""
+        self.assertEqual(self.castle.name, "Test Castle")
+
+    def test_castle_boss(self):
+        """Test that the default boss is Bowser."""
+        self.assertEqual(self.castle.boss, "Bowser")
+
+    def test_castle_world(self):
+        """Test that the default world is Grass Land."""
+        self.assertEqual(self.castle.world, "Grass Land")
+
+    def test_has_access_granted(self):
+        """Test that access is granted for the correct powerup."""
+        character = Mock(powerup="Super Mushroom")
+        self.assertTrue(self.castle.has_access(character))
+
+    def test_has_access_denied(self):
+        """Test that access is denied for an incorrect powerup."""
+        character = Mock(powerup="Starman")
+        self.assertFalse(self.castle.has_access(character))
+
+    def test_empty_name_raises_error(self):
+        """Test that an empty castle name raises a ValueError."""
+        with self.assertRaises(ValueError):
+            Castle("")
+
+
+class TestCharacter(unittest.TestCase):
+    """Tests for the Character class."""
+
+    def setUp(self):
+        """Set up a test character."""
+        self.character = Character("Mario")
+
+    def test_character_name(self):
+        """Test that the character name is set correctly."""
+        self.assertEqual(self.character.name, "Mario")
+
+    def test_default_powerup(self):
+        """Test that the default powerup is None."""
+        self.assertIsNone(self.character.powerup)
+
+    def test_set_powerup(self):
+        """Test setting a powerup."""
+        self.character.powerup = "Fire Flower"
+        self.assertEqual(self.character.powerup, "Fire Flower")
+
+    def test_empty_name_raises_error(self):
+        """Test that an empty character name raises a ValueError."""
+        with self.assertRaises(ValueError):
+            Character("")
+
+
+class TestCastleAndCharacter(unittest.TestCase):
+    """Tests for the interaction between Castle and Character."""
+
+    def setUp(self):
+        """Set up a test castle and character."""
+        self.castle = Castle("Test Castle")
+        self.character = Character("Mario")
+
+    def test_character_has_access(self):
+        """Test that a character with the correct powerup has access."""
+        self.character.powerup = "Super Mushroom"
+        self.assertTrue(self.castle.has_access(self.character))
+
+    def test_character_denied_access(self):
+        """Test that a character with the wrong powerup is denied access."""
+        self.character.powerup = "Starman"
+        self.assertFalse(self.castle.has_access(self.character))
+
+
+if __name__ == "__main__":
+    unittest.main()

unittest_to_pytest/input_repo/run_tests.py

@@ -2,10 +2,9 @@
 from codegen import Codebase
 
 # Initialize codebase
-codebase = Codebase("./")
 
 # Define the target directory
-TARGET_DIR = "repo-before/tests"
+TARGET_DIR = "input_repo/tests"
 
 
 def remove_unittest_inheritance(file):
@@ -78,4 +77,5 @@ def run(codebase: Codebase):
 
 
 if __name__ == "__main__":
+    codebase = Codebase("./")
     run(codebase)