Update test_code.py

Author: eshaan-deepsource

tests/test_code.py

@@ -9,3 +9,6 @@ def test_random_number_generator():
 class Tests(unittest.TestCase):
     def my_test(self, arg1, arg2):
         self.assertEquals(arg1, arg2)
+
+    def my_test2(self, arg1, arg2):
+        self.assertEquals(arg1, arg2)