Some tests in PartitionParserTests were failing intermittently due to the usage of non-synchronized shared state between concurrent threads. This commit updates the test code to use `AtomicInteger` instead of `int` for the state shared between concurrent threads.