Python Stimulus2D Examples

Example #1

File: test_stimulus.py Project: jhamrick/optimal-mental-rotation

def test_random():
    stim = Stimulus2D.random(8)
    assert stim._v.shape[0] == 8

    for i in xrange(20):
        stim = Stimulus2D.random((6, 8))
        n = stim._v.shape[0]
        assert n >= 6 and n <= 8

    with pytest.raises(ValueError):
        stim = Stimulus2D.random((8, 6))

    with pytest.raises(ValueError):
        stim = Stimulus2D.random((8, 6, 8))

    with pytest.raises(ValueError):
        stim = Stimulus2D.random((8,))

Example #2

File: test_stimulus.py Project: jhamrick/optimal-mental-rotation

def test_io(X0):
    fh = NamedTemporaryFile()
    X0.save(fh.name, force=True)
    X1 = Stimulus2D.load(fh.name)
    assert X0 == X1