matlab `fread` in python with numpy

matlab_fread_in_python.py

from typing import Union, BinaryIO

import numpy as np

from binary_reader import BinaryReader


def f_read(reader: Union[BinaryIO, BinaryReader], shape, dtype):
    """
    ``fread`` method in matlab
    """
    size = int(np.iinfo(dtype).bits / 8)

    if isinstance(shape, (list, tuple)):
        count = shape[0] * shape[1]
        b = reader.read(int(size * count))
        return np.frombuffer(b, dtype, count=count).reshape(shape, order="F")

    return np.frombuffer(reader.read(size * shape), dtype, count=shape)