pytest_databricks.py

# Databricks notebook source
# MAGIC %pip install -r requirements.txt
# MAGIC %pip install pytest

# COMMAND ----------

dbutils.library.restartPython()

# COMMAND ----------

import logging
logging.getLogger('py4j.java_gateway').setLevel(logging.ERROR)

# COMMAND ----------

import dbx
import pytest
import sys
import os

sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath("../"+dbx.__file__))))

os.environ['DATABRICKS_HOST'] = dbutils.notebook.entry_point.getDbutils().notebook().getContext().apiUrl().get()
os.environ['DATABRICKS_TOKEN'] = dbutils.notebook.entry_point.getDbutils().notebook().getContext().apiToken().get()

sys.dont_write_bytecode = True

result = pytest.main(['--import-mode=importlib', 'tests/dbx/'])
if result != 0:
    raise Exception(f"Tests failed with exit code: {result}")