test_class.declarator: affinityFunction
test_class.modifier: @Mock
      private
test_class.original_string: @Mock
      private AffinityFunction affinityFunction;
test_class.type: AffinityFunction
test_class.var_name: affinityFunction
test_class.declarator: wrapper
test_class.modifier: private
test_class.original_string: private DatasetAffinityFunctionWrapper wrapper;
test_class.type: DatasetAffinityFunctionWrapper
test_class.var_name: wrapper
test_class.file: modules/ml/src/test/java/org/apache/ignite/ml/dataset/impl/cache/util/DatasetAffinityFunctionWrapperTest.java
test_class.identifier: DatasetAffinityFunctionWrapperTest
test_case.body: @Test
      public void testPartitions() {
          doReturn(42).when(affinityFunction).partitions();
  
          int partitions = wrapper.partitions();
  
          assertEquals(42, partitions);
          verify(affinityFunction, times(1)).partitions();
      }
test_case.class_method_signature: DatasetAffinityFunctionWrapperTest.testPartitions()
test_case.full_signature: @Test public void testPartitions()
test_case.identifier: testPartitions
test_case.invocations: partitions when doReturn partitions assertEquals partitions verify times
test_case.modifiers: @Test public
test_case.parameters: ()
test_case.return: void
test_case.signature: void testPartitions()
focal_class.declarator: serialVersionUID = -8233787063079973753L
focal_class.modifier: private static final
focal_class.original_string: private static final long serialVersionUID = -8233787063079973753L;
focal_class.type: long
focal_class.var_name: serialVersionUID
focal_class.declarator: delegate
focal_class.modifier: private final
focal_class.original_string: private final AffinityFunction delegate;
focal_class.type: AffinityFunction
focal_class.var_name: delegate
focal_class.file: modules/ml/src/main/java/org/apache/ignite/ml/dataset/impl/cache/util/DatasetAffinityFunctionWrapper.java
focal_class.identifier: DatasetAffinityFunctionWrapper
focal_class.interfaces: implements AffinityFunction
focal_class.class_method_signature: DatasetAffinityFunctionWrapper.DatasetAffinityFunctionWrapper(AffinityFunction delegate)
focal_class.full_signature: public  DatasetAffinityFunctionWrapper(AffinityFunction delegate)
focal_class.identifier: DatasetAffinityFunctionWrapper
focal_class.modifiers: public
focal_class.parameters: (AffinityFunction delegate)
focal_class.signature:  DatasetAffinityFunctionWrapper(AffinityFunction delegate)
focal_class.class_method_signature: DatasetAffinityFunctionWrapper.reset()
focal_class.full_signature: @Override public void reset()
focal_class.identifier: reset
focal_class.modifiers: @Override public
focal_class.parameters: ()
focal_class.return: void
focal_class.signature: void reset()
focal_class.class_method_signature: DatasetAffinityFunctionWrapper.partitions()
focal_class.full_signature: @Override public int partitions()
focal_class.identifier: partitions
focal_class.modifiers: @Override public
focal_class.parameters: ()
focal_class.return: int
focal_class.signature: int partitions()
focal_class.class_method_signature: DatasetAffinityFunctionWrapper.partition(Object key)
focal_class.full_signature: @Override public int partition(Object key)
focal_class.identifier: partition
focal_class.modifiers: @Override public
focal_class.parameters: (Object key)
focal_class.return: int
focal_class.signature: int partition(Object key)
focal_class.class_method_signature: DatasetAffinityFunctionWrapper.assignPartitions(AffinityFunctionContext affCtx)
focal_class.full_signature: @Override public List<List<ClusterNode>> assignPartitions(AffinityFunctionContext affCtx)
focal_class.identifier: assignPartitions
focal_class.modifiers: @Override public
focal_class.parameters: (AffinityFunctionContext affCtx)
focal_class.return: List<List<ClusterNode>>
focal_class.signature: List<List<ClusterNode>> assignPartitions(AffinityFunctionContext affCtx)
focal_class.class_method_signature: DatasetAffinityFunctionWrapper.removeNode(UUID nodeId)
focal_class.full_signature: @Override public void removeNode(UUID nodeId)
focal_class.identifier: removeNode
focal_class.modifiers: @Override public
focal_class.parameters: (UUID nodeId)
focal_class.return: void
focal_class.signature: void removeNode(UUID nodeId)
focal_method.body: @Override public int partitions() {
          return delegate.partitions();
      }
focal_method.class_method_signature: DatasetAffinityFunctionWrapper.partitions()
focal_method.full_signature: @Override public int partitions()
focal_method.identifier: partitions
focal_method.invocations: partitions
focal_method.modifiers: @Override public
focal_method.parameters: ()
focal_method.return: int
focal_method.signature: int partitions()