Code to "test" openpyxl pandas xlsx export

make_or_break_xlsx.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
import json
import numpy as np
import openpyxl
import pandas as pd
import string
import sys


def main():
    """Code to "test" openpyxl pandas xlsx export"""
    filename = "text.xlsx"
    limit = 2 ** 8

    data = {}
    cols = []
    for i in range(limit):
        cols.append(make_label())

    for j in range(limit):
        line = {}
        for col in cols:
            line[col] = np.random.random()
        data[make_label()] = line
    try:
        frame = pd.DataFrame.from_dict(data)
        frame.to_excel(filename)
    except:
        frame.to_json("dump.json")
        raise
    print "Test passed, openpyxl version:", openpyxl.__version__
                

def make_label():
    """generate lables for attempts to cause errors"""
    name_letters = string.lowercase + string.uppercase + "_:/% "
    return string.lowercase[np.random.choice(len(string.lowercase))] + \
         "".join([name_letters[np.random.choice(len(name_letters))]
                  for zed in range(11)])

    
if __name__ == '__main__':
    main()