Pretty Main

pretty_main.py

import classical
import quantum
import pandas as pd
from IPython.core import display as ICD

sbits = [0, 1]
tbits = [(0, 0), (0, 1), (1, 0), (1, 1)]
tables = []

classicalTable = pd.DataFrame(columns=['bit', 'not'])
quantumTable = pd.DataFrame(columns=['Qbit', 'not'])
for bit in sbits:
    classicalTable = classicalTable.append({'bit': bit, 
                                            'not': classical.not_gate(bit)}, 
                                            ignore_index=True)
    quantumTable = quantumTable.append({'Qbit': "|" + str(bit) + "⟩", 
                                        'not': "|" + str(quantum.not_gate(bit)) + "⟩"}, 
                                        ignore_index=True)
print('Gate', 'not')
ICD.display(classicalTable)
ICD.display(quantumTable)

gates = [('and', classical.and_gate, quantum.and_gate),
         ('or', classical.or_gate, quantum.or_gate),
         ('xor', classical.xor_gate, quantum.xor_gate)]

for i in range(len(gates)):
    classicalTable = pd.DataFrame(columns=['bit1', 'bit2', gates[i][0]])
    quantumTable = pd.DataFrame(columns=['Qbit1', 'Qbit2', gates[i][0]])
    for bit in tbits:
        classicalTable = classicalTable.append({'bit1': bit[0], 
                                                'bit2': bit[1], 
                                                gates[i][0]: gates[i][1](bit[0], 
                                                                         bit[1])}, 
                                                ignore_index=True)
        quantumTable = quantumTable.append({'Qbit1': "|" + str(bit[0]) + "⟩", 
                                            'Qbit2': "|" + str(bit[1]) + "⟩", 
                                            gates[i][0]: "|" + str(gates[i][2](bit[0], 
                                                                     bit[1])) + "⟩"}, 
                                            ignore_index=True)
    print('Gate', gates[i][0])
    ICD.display(classicalTable)
    ICD.display(quantumTable)