import firehole as fh
import numpy as np


li = [[100, 90, 100, 84, 90, 100, 100, 100, 100],
      [100, 100, 78.6, 100, 90, 100, 100, 100, 100],
      [75, 100, 85.7, 100, 90, 100, 100, 100, 100],
      [100, 100, 78.6, 100, 90, 100, 94.4, 100, 100],
      [100, 90, 100, 100, 100, 90, 100, 100, 80],
      [100, 100, 100, 100, 90, 100, 100, 85.7, 100],
      [100, 100, 78.6, 100, 90, 100, 55.6, 100, 100],
      [87.5, 100, 85.7, 100, 100, 100, 100, 100, 100],
      [100, 100, 92.9, 100, 80, 100, 100, 100, 100],
      [100, 90, 100, 100, 100, 100, 100, 100, 100],
      [100, 100, 92.9, 100, 90, 100, 100, 100, 100]]

# convert list into numpy array
li = np.array(li)
# calculation by entropy method
entropy = fh.Entropy()
rlt = entropy.entropy(li)

print(rlt['entropy'])
print(rlt['weight'])
print(rlt['score'])
