• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• aboleth-master
  • .circleci
    • config.yml
  • Makefile
  • LICENSE
  • aboleth
    • hlayers.py
    • util.py
    • distributions.py
    • prediction.py
    • version.py
    • random.py
    • impute.py
    • losses.py
    • datasets.py
    • layers.py
    • __init__.py
    • kernels.py
    • baselayers.py
    • initialisers.py
  • README.rst
  • setup.py
  • data
    • motorcycle_Y.npy
    • motorcycle_X.npy
  • AUTHORS.rst
  • demos
    • mnist_softmax_regression.py
    • imputation.py
    • multi_input.py
    • classification.py
    • regression.py
    • sarcos.py
    • regression_keras.py
    • regression_tutorial.py
  • setup.cfg
  • tests
    • test_utils.py
    • util.py
    • test_layers.py
    • test_prediction.py
    • test_hlayers.py
    • __init__.py
    • test_kernels.py
    • test_impute.py
    • test_distributions.py
    • test_initialisers.py
    • test_losses.py
    • test_baselayers.py
    • conftest.py
  • .gitignore
  • docs
    • Makefile
    • api
      • distributions.rst
      • datasets.rst
      • baselayers.rst
      • losses.rst
      • layers.rst
      • util.rst
      • api.rst
      • impute.rst
      • random.rst
      • kernels.rst
      • hlayers.rst
    • tutorials
      • saving_loading.rst
      • regression_figs
      • some_regressors.rst
      • tutorials.rst
      • keras_integration.rst
    • demos.rst
    • authors.rst
    • quickstart.rst
    • requirements.txt
    • index.rst
    • conf.py
    • contributing.rst
    • installation.rst
  • MANIFEST.in
  • RELEASE.rst

"""Test the aboleth utilities."""

from types import GeneratorType

import numpy as np
import tensorflow as tf

import aboleth as ab


def test_batch():
    """Test the batch feed dict generator."""
    X = np.arange(100)
    fd = {'X': X}

    data = ab.batch(fd, batch_size=10, n_iter=10)

    # Make sure this is a generator
    assert isinstance(data, GeneratorType)

    # Make sure we get a dict back of a length we expect
    d = next(data)
    assert isinstance(d, dict)
    assert 'X' in d
    assert len(d['X']) == 10

    # Test we get all of X back in one sweep of the data
    accum = list(d['X'])
    for ds in data:
        assert len(ds['X']) == 10
        accum.extend(list(ds['X']))

    assert len(accum) == len(X)
    assert set(X) == set(accum)


def test_batch_predict():
    """Test the batch prediction feed dict generator."""
    X = np.arange(100)
    fd = {'X': X}

    data = ab.batch_prediction(fd, batch_size=10)

    # Make sure this is a generator
    assert isinstance(data, GeneratorType)

    # Make sure we get a dict back of a length we expect with correct indices
    for ind, d in data:
        assert isinstance(d, dict)
        assert 'X' in d
        assert len(d['X']) == 10
        assert all(X[ind] == d['X'])