Effectuer un test unitaire de Databricks Notebook

introduction

Effectuez un test unitaire du portable suivant

MyNotebook

def hoge(i):
  return 'hoge'*i

def fuga(i):
  return 'fuga'*i

Créer un cahier de test

Créez un cahier de testMyNotebookTest dans le même dossier que NotebookMyNotebook

MyNotebookTest

# Cmd1 
%run "./MyNotebook"

# Cmd2
import unittest

class MyNotebookTests(unittest.TestCase):
  
  def test_hoge(self):
    self.assertEqual(hoge(3), 'hogehogehoge')
    self.assertNotEqual(hoge(2), 'hoge')

  def test_fuga(self):
    self.assertEqual(fuga(3), 'fugafugafuga')
    self.assertNotEqual(fuga(2), 'fuga')

suite = unittest.TestLoader().loadTestsFromTestCase(MyNotebookTests)
runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite)

Comme point de créer

• Exécutez d'abord la commande % run avec Cmd 1 pour exécuter le notebook sous test.

%run "./MyUnittest"

• Ecrire un cas de test en utilisant unittest qui est un framework de test python dans Cmd 2 ou version ultérieure.

Résultat d'exécution

Confirmez que le test est OK comme suit

test_fuga (__main__.MyNotebookTests) ... ok
test_hoge (__main__.MyNotebookTests) ... ok

----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK
Out[50]: <unittest.runner.TextTestResult run=2 errors=0 failures=0>