test.py

#!/usr/bin/env python3
from simulation import Simulation

def testcase1(sim):
    sim.enable_cell(1, 0, 100)
    sim.enable_cell(0, 1, 0)
    sim.enable_cell(1, 1, 0)
    sim.enable_cell(2, 1, 0)
    sim.hor_v[0][1] = 100

def testcase2(sim):
    sim.enable_cell(1, 0, 100)
    sim.enable_cell(0, 1, 60)
    sim.enable_cell(1, 1, 0)
    sim.enable_cell(2, 1, 0)
    sim.hor_v[0][1] = 100

def testcase3(sim):
    sim.enable_cell(1, 0, 100)
    sim.enable_cell(0, 1, 100)
    sim.enable_cell(1, 1, 0)
    sim.enable_cell(2, 1, 0)
    sim.hor_v[0][1] = 100
    sim.ver_v[1][0] = 100

def testcase4(sim):
    sim.enable_cell(1, 0, 100)
    sim.enable_cell(0, 1, 20)
    sim.enable_cell(1, 1, 20)
    sim.enable_cell(2, 1, 20)
    sim.hor_v[0][1] = 100
    sim.ver_v[1][0] = 20
    sim.ver_v[1][1] = 20
    sim.ver_v[1][2] = 20
    sim.ver_v[1][3] = 20

def testcase5():
    sim = Simulation(10, 1)
    for i in range(10):
        sim.enable_cell(i, 0, 0)
    sim.ver_v[0][0] = 100
    sim.ver_v[0][10] = 100
    return sim

def testcase6(sim):
    sim.enable_cell(1, 0, 100)
    sim.enable_cell(0, 1, 0)
    sim.enable_cell(1, 1, 0)
    sim.enable_cell(2, 1, 0)
    sim.hor_v[0][1] = 25
    sim.ver_v[1][0] = -10
    sim.ver_v[1][3] = 20


s = Simulation(3, 2)
testcase6(s)
#s = testcase5()

s.update_display()

s.display.print()

while True:
    inp = input()
    if inp == 'q':
        break
    print("\r\033[1A", end="")
    s.display.clean()
    s.step()
    s.display.print()