from more_itertools import divide

import bpytop
from bpytop import (CORES, SYSTEM, THREADS, Fx, create_box, floating_humanizer,
                    get_cpu_core_mapping, get_cpu_name, units_to_bytes)


def test_get_cpu_name():
	assert isinstance(get_cpu_name(), str)

def test_get_cpu_core_mapping():
	cpu_core_mapping = get_cpu_core_mapping()
	assert isinstance(cpu_core_mapping, list)
	# Assert cpu submappings are sequential
	for submapping in divide(THREADS//CORES, cpu_core_mapping):
		submapping = list(submapping)
		for a, b in zip(submapping[:-1], submapping[1:]):
			assert b - a == 1

def test_create_box():
	assert len(create_box(x=1, y=1, width=10, height=10, title="", title2="", line_color=None, title_color=None, fill=True, box=None)) > 1

def test_floating_humanizer():
	assert floating_humanizer(100) == "100 Byte"
	assert floating_humanizer(100<<10) == "100 KiB"
	assert floating_humanizer(100<<20, bit=True) == "800 Mib"
	assert floating_humanizer(100<<20, start=1) == "100 GiB"
	assert floating_humanizer(100<<40, short=True) == "100T"
	assert floating_humanizer(100<<50, per_second=True) == "100 PiB/s"

def test_units_to_bytes():
	assert units_to_bytes("10kbits") == 1280
	assert units_to_bytes("100Mbytes") == 104857600
	assert units_to_bytes("1gbit") == 134217728