old_contents: # -*- coding: utf-8 -*-
  from __future__ import (absolute_import, division,
                          print_function, unicode_literals)
  from builtins import (  # pylint: disable=redefined-builtin, unused-import
      bytes, dict, int, list, object, range, str,
      ascii, chr, hex, input, next, oct, open,
      pow, round, super,
      filter, map, zip)
  import os
  import ndd
  import pytest
  import numpy
  from numpy import random as random
  
  EPS = 0.0001
  
  def tests_dir():
      """Return None is no tests dir."""
      cwd = os.getcwd()
      basename = os.path.basename(cwd)
      if basename == 'tests':
          return cwd
      else:
          tests_dir = os.path.join(cwd, 'tests')
          if os.path.exists(tests_dir):
              return tests_dir
  
  def random_counts(n=None, k=None, alpha=None):
      random.seed(123)
      pp = random.dirichlet([alpha]*k)
      return random.multinomial(n, pp)
  
  tdir = tests_dir()
  import json
  with open(os.path.join(tdir, 'data.json'), 'r') as _jf:
      CASES = json.load(_jf)
  
  @pytest.mark.parametrize('setting, kwargs, result', CASES)
  def test_ndd(setting, kwargs, result):
      """Basic tests."""
      counts = random_counts(**setting)
      test_result = ndd.entropy(counts, k=setting['k'], **kwargs)
      assert numpy.abs(test_result - numpy.float64(result)) < EPS
  
new_contents: # -*- coding: utf-8 -*-
  from __future__ import (absolute_import, division,
                          print_function, unicode_literals)
  from builtins import (  # pylint: disable=redefined-builtin, unused-import
      bytes, dict, int, list, object, range, str,
      ascii, chr, hex, input, next, oct, open,
      pow, round, super,
      filter, map, zip)
  import os
  import json
  import pytest
  import numpy
  from numpy import random as random
  import ndd
  
  EPS = 0.0001
  
  
  def tests_dir():
      """Return None is no tests dir."""
      cwd = os.getcwd()
      basename = os.path.basename(cwd)
      if basename == 'tests':
          return cwd
      else:
          tests_dir = os.path.join(cwd, 'tests')
          if os.path.exists(tests_dir):
              return tests_dir
  
  
  def random_counts(n=None, k=None, alpha=None):
      random.seed(123)
      pp = random.dirichlet([alpha]*k)
      return random.multinomial(n, pp)
  
  
  with open(os.path.join(tests_dir(), 'data.json'), 'r') as _jf:
      CASES = json.load(_jf)
  
  
  @pytest.mark.parametrize('setting, kwargs, result', CASES)
  def test_ndd(setting, kwargs, result):
      """Basic tests."""
      counts = random_counts(**setting)
      test_result = ndd.entropy(counts, k=setting['k'], **kwargs)
      assert numpy.abs(test_result - numpy.float64(result)) < EPS
  
subject: fix style