package org.nd4j.linalg.ops.transforms;

import static org.junit.Assert.*;

import org.junit.Test;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.factory.Nd4j;


/**
 * Created by agibsonccc on 9/6/14.
 */
public abstract class TransformTests {



    @Test
    public void testPooling() {
        INDArray twoByTwo = Nd4j.ones(new int[]{2,2,2});
        INDArray pool = Transforms.pool(twoByTwo,new int[]{1,2});

    }

    @Test
    public void testMaxPooling() {
        INDArray nd = Nd4j.rand(new int[]{1,2,3,4});
        INDArray pool = Transforms.maxPool(nd,new int[]{1,2},false);
        pool = Transforms.maxPool(nd,new int[]{1,2},true);
    }



    @Test
    public void testPow() {
        INDArray twos = Nd4j.valueArrayOf(2,2);
        INDArray pow = Transforms.pow(twos,2);
        assertEquals(Nd4j.valueArrayOf(2,4),pow);
    }



}