Python unitsum_tuples Examples

Example #1

File: test_combinatorics.py Project: Autoplectic/dit

def test_unitsum_tuples2():
    s = np.asarray(list(unitsum_tuples(4, 3, -1, 3)))
    s_ = np.asarray([
        (-1.0, -1.0, 3.0),
        (-1.0, 0.0, 2.0),
        (-1.0, 1.0, 1.0),
        (-1.0, 2.0, 0.0),
        (-1.0, 3.0, -1.0),
        (0.0, -1.0, 2.0),
        (0.0, 0.0, 1.0),
        (0.0, 1.0, 0.0),
        (0.0, 2.0, -1.0),
        (1.0, -1.0, 1.0),
        (1.0, 0.0, 0.0),
        (1.0, 1.0, -1.0),
        (2.0, -1.0, 0.0),
        (2.0, 0.0, -1.0),
        (3.0, -1.0, -1.0),
    ])
    assert np.allclose(s, s_)

Example #2

def test_unitsum_tuples2():
    s = np.asarray(list(unitsum_tuples(4, 3, -1, 3)))
    s_ = np.asarray([
        (-1.0, -1.0, 3.0),
        (-1.0, 0.0, 2.0),
        (-1.0, 1.0, 1.0),
        (-1.0, 2.0, 0.0),
        (-1.0, 3.0, -1.0),
        (0.0, -1.0, 2.0),
        (0.0, 0.0, 1.0),
        (0.0, 1.0, 0.0),
        (0.0, 2.0, -1.0),
        (1.0, -1.0, 1.0),
        (1.0, 0.0, 0.0),
        (1.0, 1.0, -1.0),
        (2.0, -1.0, 0.0),
        (2.0, 0.0, -1.0),
        (3.0, -1.0, -1.0),
    ])
    assert np.allclose(s, s_)

Example #3

def test_unitsum_tuples3():
    # Violate: 1 = mx + (k-1) * mn
    g = unitsum_tuples(3, 3, 1, 0)
    with pytest.raises(Exception):
        next(g)

Example #4

def test_unitsum_tuples1():
    s = np.asarray(list(unitsum_tuples(3, 2, .2, .8)))
    s_ = np.asarray([[.2, .8], [.4, .6], [.6, .4], [.8, .2]])
    assert np.allclose(s, s_)

Example #5

File: test_combinatorics.py Project: Autoplectic/dit

def test_unitsum_tuples3():
     # Violate: 1 = mx + (k-1) * mn
     g = unitsum_tuples(3, 3, 1, 0)
     with pytest.raises(Exception):
         next(g)

Example #6

File: test_combinatorics.py Project: Autoplectic/dit

def test_unitsum_tuples1():
    s = np.asarray(list(unitsum_tuples(3, 2, .2, .8)))
    s_ = np.asarray([[.2, .8], [.4, .6], [.6, .4], [.8, .2]])
    assert np.allclose(s, s_)

Example #7

def test_unitsum_tuples3():
    # Violate: 1 = mx + (k-1) * mn
    g = unitsum_tuples(3, 3, 1, 0)
    assert_raises(Exception, next, g)

Example #8

File: test_combinatorics.py Project: chebee7i/dit

def test_unitsum_tuples3():
     # Violate: 1 = mx + (k-1) * mn
     g = unitsum_tuples(3, 3, 1, 0)
     assert_raises(Exception, next, g)