Test case generator for https://github.com/subsr97/daily-coding-problem/tree/master/challenges/busiest-time-in-building.py

busiest-time-in-building-test.py

from time import time
import random

def getTimestamp():
    return int(time())

entries = []
people = 0
peopleList = [0]
timestamp = getTimestamp()

while True:
    d = dict()
    timestamp += random.randint(0,120)
    d["timestamp"] = timestamp
    d["count"] = random.randint(1,5)
    d["type"] = random.choice(["enter", "exit"])
    if d["type"] == "exit":
        if people - d["count"] < 0:
            continue
        else:
            people -= d["count"]
    else:
        people += d["count"]
    entries.append(d)
    peopleList.append(people)
    if people == 0:
        break

print("Entry Count:", len(entries))
print("Entries:\n", entries)
print("People List:", peopleList)
print("Answer:", (entries[peopleList.index(max(peopleList))-1]["timestamp"], entries[peopleList.index(max(peopleList))]["timestamp"]))