test/gensample.py

   1 #!/usr/bin/env python
   2
   3 import sys
   4 import random
   5 import string
   6
   7
   8 f1 = open(sys.argv[1])
   9 f2 = open(sys.argv[2])
  10 count = int(sys.argv[3])
  11
  12 l1 = f1.readlines()
  13 l2 = f2.readlines()
  14
  15 l1 = [i.strip() for i in l1]
  16 l2 = [i.strip() for i in l2]
  17
  18 d1 = {}
  19 d2 = {}
  20
  21 # inicializo todo vacio
  22 for p in range(0, count):
  23         d1[l1[p]] = []
  24         d2[l2[p]] = []
  25
  26 # agrego a los de d1 las claves de d2, mezclandola al azar
  27 l = d2.keys()
  28 for i in d1.keys():
  29         random.shuffle(l)
  30         d1[i] = l[:]
  31
  32 # y lo mismo para el otro
  33 l = d1.keys()
  34 for i in d2.keys():
  35         random.shuffle(l)
  36         d2[i] = l[:]
  37
  38 # imprimo
  39 for i in d1.keys():
  40         print "%s: %s" % (i, string.join(d1[i], ','))
  41 print
  42 for i in d2.keys():
  43         print "%s: %s" % (i, string.join(d2[i], ','))
  44
  45