Added equivalence partition and unit tests