["--width", "512", "--height", "512"], ["--width", "1024", "--height", "1024"], ["--width", "2048", "--height", "2048"] ] SIMPLEBIG_ARGS = list(product(COMPLEXITIES, DATASIZES, BORDERS)) #################### # Device Arguments # #################### if system.HOSTNAME == "cec": DEVARGS = [["--device-type", "CPU", "--device-count", "1"]] elif system.HOSTNAME == "dhcp-90-060": DEVARGS = [["--device-type", "GPU", "--device-count", "1"]] elif system.HOSTNAME == "florence": DEVARGS = [["--device-type", "CPU", "--device-count", "1"]] elif system.HOSTNAME == "monza": DEVARGS = [["--device-type", "CPU", "--device-count", "1"], ["--device-type", "GPU", "--device-count", "1"], ["--device-type", "GPU", "--device-count", "2"]] elif system.HOSTNAME == "tim": DEVARGS = [["--device-type", "CPU", "--device-count", "1"], ["--device-type", "GPU", "--device-count", "1"], ["--device-type", "GPU", "--device-count", "2"], ["--device-type", "GPU", "--device-count", "3"], ["--device-type", "GPU", "--device-count", "4"]] elif system.HOSTNAME == "whz5": DEVARGS = [["--device-type", "GPU", "--device-count", "1"]] else: io.warn("Unrecognised hostname, no devargs.")
DATASIZES = [["--width", "512", "--height", "512"], ["--width", "1024", "--height", "1024"], ["--width", "2048", "--height", "2048"]] SIMPLEBIG_ARGS = list(product(COMPLEXITIES, DATASIZES, BORDERS)) #################### # Device Arguments # #################### if system.HOSTNAME == "cec": DEVARGS = [["--device-type", "CPU", "--device-count", "1"]] elif system.HOSTNAME == "dhcp-90-060": DEVARGS = [["--device-type", "GPU", "--device-count", "1"]] elif system.HOSTNAME == "florence": DEVARGS = [["--device-type", "CPU", "--device-count", "1"]] elif system.HOSTNAME == "monza": DEVARGS = [["--device-type", "CPU", "--device-count", "1"], ["--device-type", "GPU", "--device-count", "1"], ["--device-type", "GPU", "--device-count", "2"]] elif system.HOSTNAME == "tim": DEVARGS = [["--device-type", "CPU", "--device-count", "1"], ["--device-type", "GPU", "--device-count", "1"], ["--device-type", "GPU", "--device-count", "2"], ["--device-type", "GPU", "--device-count", "3"], ["--device-type", "GPU", "--device-count", "4"]] elif system.HOSTNAME == "whz5": DEVARGS = [["--device-type", "GPU", "--device-count", "1"]] else: io.warn("Unrecognised hostname, no devargs.")
def test_warn(): out = StringIO() io.warn("foo", file=out) assert "WARN" == re.search("WARN", out.getvalue()).group(0)
def test_warn(self): out = StringIO() io.warn("foo", file=out) self._test("WARN", re.search("WARN", out.getvalue()).group(0))