Let $f : A \rightarrow B$ is the statement $f$ is a bijection if and only if for each partition $\left\{A_j \right\}_{j \in J}$ then $\left\{ f(A_j) \right\}_{j\in J}$ is a partition for $B$ true?
I think it is because... if $f$ is a bijection then $\forall x,y \in A, x \neq y \Rightarrow f(x) \neq f(y)$, I can then conclude then $\forall i,j, i \neq j, f(A_i) \neq f(A_j)$ and then $B = f\left(\bigcup_{j\in J} A_j \right) = \bigcup_{j\in J} f(A_j)$, here $\left\{ A_j \right\}_{j \in J}$ is a partition for $A$. Conversely instead I can consider the partition formed by sets of the form
$$ A_x = \left\{ x \right\} $$
where $x \in A$. So $f(A_x) = \left\{f(x)\right\}$ and $\bigcup_{x \in A} f(A_x) = B$ with $f(A_x) \cap f(A_y) = \emptyset$ when $x \neq y, \Rightarrow f(x) \neq f(y)$. Can the statement be restricted to something like $f$ is a bijection if and only if there is a partition of $A$ such that $\left\{ f(A_j) \right\}_{j \in J}$ is a partition for $B$, with $f|_{A_j}$ bijection itself? I think this is also true since:
The case $f$ is a bijection then there's a partition $\left\{ A_j \right\}_{j \in J}$ such that $f |_{A_j}$ is a bijection is easy to prove by just considering the partition which elements are $A_x = \left\{ x \right\}$, $x \in A$. Conversely let's consider a partition $\left\{ A_j \right\}_{j \in J}$ with such features, let's pick $x,y \in A$ such that $x \neq y$ then there are two indices $j_x,j_y \in J$ such that $x \in A_{j_x}$ and $y \in A_{j_y}$. If $j_x = j_y$ since the restriction to $A_{j_x}$ is a bijection then $f(x) \neq f(y)$, otherwise since by hypothesis $\left\{ f(A_j) \right\}_{j \in J}$ is a partition for $B$ then $f(A_{j_x}) \cap f(A_{j_y}) = \emptyset$ and then $f(x) \neq f(y)$ so $f$ is a bijection.