import unittest class TestAdd(unittest.TestCase): def test_add(self): self.assertEqual(add(1, 2), 3) def test_run_code(self): self.assertEqual(run_code('ls -l'), 'total 0\n-rw-r--r-- 1 564839 eng 0 May 28 08:56 README.md\n-rw-r--r-- 1 564839 eng 0 May 28 08:56 script.py\n-rw-r--r-- 1 564839 eng 121 May 28 08:56 test.py\n')

import unittest

def add(a, b):
  return a + b

class TestAdd(unittest.TestCase):
  def test_add(self):
    self.assertEqual(add(1, 2), 3)
  def test_run_code(self):
    self.assertEqual(run_code('ls -l'), 'total 0\n-rw-r--r--  1 564839  eng  0 May 28 08:56 README.md\n-rw-r--r--  1 564839  eng  0 May 28 08:56 script.py\n-rw-r--r--  1 564839  eng  121 May 28 08:56 test.py\n') 

if __name__ == '__main__':
  unittest.main()