def test_serialization_nonempty(self) -> None:
e = MemoryClassificationElement('test', 0)
e.set_classification(a=0, b=1)
expected_map = {'a': 0, 'b': 1}
assert e._c == expected_map
e2 = pickle.loads(pickle.dumps(e))
assert e2._c == expected_map
def test_set_classification(self) -> None:
"""
Test setting valid classification map.
"""
e = MemoryClassificationElement('test', 0)
expected_map = {'a': 1, 'b': 0}
e.set_classification(expected_map)
assert e._c == expected_map