code: def test_not_implemented():
      """Test if all NotImplementedError are raise correctly"""
      pd = ProbabilityDistribution()
  
      with pytest.raises(NotImplementedError):
          pd.log_prob(None)
  
      with pytest.raises(NotImplementedError):
          pd.entropy()
  
      with pytest.raises(NotImplementedError):
          pd.kl(ProbabilityDistribution())
  
      with pytest.raises(NotImplementedError):
          pd.sample()
  
      with pytest.raises(NotImplementedError):
          pd.deterministic_sample()
  
file_path: maze/maze/test/distributions/test_distribution.py
imports: import numpy
  import pytest
  import torch
  from gym import spaces
  from maze.distributions.bernoulli import BernoulliProbabilityDistribution
  from maze.distributions.distribution import ProbabilityDistribution
label_desc: Use image node_id 2 for calling a global function with example usage: test_not_implemented() without return types