adding test to find code

Author: zhenningdavidliu

src/inspect_evals/class_eval/test.py

@@ -1,15 +1,18 @@
 import unittest
 
 from utils import construct_prompt
-from class_eval import record_to_sample
+from class_eval import record_to_sample, find_code
 
-from test_data import record, sample, prompt
+from test_data import record, sample, prompt, sample_code, raw_code
 
 class TestClassEval(unittest.TestCase):
 
     def test_record_to_sample(self):
         self.assertEqual(record_to_sample(record), sample)
 
+    def test_find_code(self):
+        self.assertEqual(find_code(raw_code), sample_code)
+
 class TestUtils(unittest.TestCase):
 
     def test_construct_prompt(self):

src/inspect_evals/class_eval/test_data.py

@@ -68,4 +68,12 @@ def add(self, x, y):
         >>> adder.add(1,2)
 
             ### Response:
-            '''
+            '''
+
+raw_code = r"""Test text
+```python
+print("Hello, World!")```
+Test text 
+"""
+
+sample_code = r"""print("Hello, World!")"""