Search in sources :

Example 1 with FreqMap

use of uk.gov.gchq.gaffer.types.FreqMap in project Gaffer by gchq.

the class FreqMapAggregatorTest method shouldMergeFreqMaps.

@Test
public void shouldMergeFreqMaps() {
    // Given
    final FreqMapAggregator aggregator = new FreqMapAggregator();
    aggregator.init();
    final FreqMap freqMap1 = new FreqMap();
    freqMap1.put("1", 2L);
    freqMap1.put("2", 3L);
    final FreqMap freqMap2 = new FreqMap();
    freqMap2.put("2", 4L);
    freqMap2.put("3", 5L);
    // When
    aggregator._aggregate(freqMap1);
    aggregator._aggregate(freqMap2);
    // Then
    final FreqMap mergedFreqMap = ((FreqMap) aggregator.state()[0]);
    assertEquals((Long) 2L, mergedFreqMap.get("1"));
    assertEquals((Long) 7L, mergedFreqMap.get("2"));
    assertEquals((Long) 5L, mergedFreqMap.get("3"));
}
Also used : FreqMap(uk.gov.gchq.gaffer.types.FreqMap) Test(org.junit.Test) AggregateFunctionTest(uk.gov.gchq.gaffer.function.AggregateFunctionTest)

Example 2 with FreqMap

use of uk.gov.gchq.gaffer.types.FreqMap in project Gaffer by gchq.

the class FreqMapSerialiserTest method shouldDeserialiseEmptyBytes.

@Override
public void shouldDeserialiseEmptyBytes() throws SerialisationException {
    // When
    final FreqMap value = serialiser.deserialiseEmptyBytes();
    // Then
    assertEquals(new FreqMap(), value);
}
Also used : FreqMap(uk.gov.gchq.gaffer.types.FreqMap)

Example 3 with FreqMap

use of uk.gov.gchq.gaffer.types.FreqMap in project Gaffer by gchq.

the class MapFilterExample method freqMapIsMoreThanOrEqualTo2.

public void freqMapIsMoreThanOrEqualTo2() {
    // ---------------------------------------------------------
    final MapFilter function = new MapFilter("key1", new IsMoreThan(2L, true));
    // ---------------------------------------------------------
    final FreqMap map1 = new FreqMap();
    map1.put("key1", 1L);
    final FreqMap map2 = new FreqMap();
    map2.put("key1", 2L);
    final FreqMap map3 = new FreqMap();
    map3.put("key1", 3L);
    final FreqMap map4 = new FreqMap();
    map4.put("key1", 3L);
    map4.put("key2", 0L);
    final FreqMap map5 = new FreqMap();
    map5.put("key2", 3L);
    runExample(function, map1, map2, map3, map4, map5);
}
Also used : FreqMap(uk.gov.gchq.gaffer.types.FreqMap) MapFilter(uk.gov.gchq.gaffer.function.MapFilter) IsMoreThan(uk.gov.gchq.gaffer.function.filter.IsMoreThan)

Example 4 with FreqMap

use of uk.gov.gchq.gaffer.types.FreqMap in project gaffer-doc by gchq.

the class FreqMapExtractorExample method multiplyAllMapValuesBy10.

public void multiplyAllMapValuesBy10() {
    // ---------------------------------------------------------
    final FreqMapExtractor function = new FreqMapExtractor("key1");
    // ---------------------------------------------------------
    final FreqMap map1 = new FreqMap();
    map1.put("key1", 1L);
    map1.put("key2", 2L);
    map1.put("key3", 3L);
    final FreqMap map2 = new FreqMap();
    map2.put("key2", 2L);
    map2.put("key3", 3L);
    final Map<String, Long> map3 = new HashMap<>();
    map3.put("key1", 1L);
    map3.put("key2", 2L);
    map3.put("key3", 3L);
    runExample(function, null, map1, map2, map3, null);
}
Also used : FreqMap(uk.gov.gchq.gaffer.types.FreqMap) HashMap(java.util.HashMap) FreqMapExtractor(uk.gov.gchq.gaffer.types.function.FreqMapExtractor)

Example 5 with FreqMap

use of uk.gov.gchq.gaffer.types.FreqMap in project gaffer-doc by gchq.

the class PredicateMapExample method freqMapIsMoreThanOrEqualTo2.

public void freqMapIsMoreThanOrEqualTo2() {
    // ---------------------------------------------------------
    final PredicateMap function = new PredicateMap("key1", new IsMoreThan(2L, true));
    // ---------------------------------------------------------
    final FreqMap map1 = new FreqMap();
    map1.put("key1", 1L);
    final FreqMap map2 = new FreqMap();
    map2.put("key1", 2L);
    final FreqMap map3 = new FreqMap();
    map3.put("key1", 3L);
    final FreqMap map4 = new FreqMap();
    map4.put("key1", 3L);
    map4.put("key2", 0L);
    final FreqMap map5 = new FreqMap();
    map5.put("key2", 3L);
    runExample(function, null, map1, map2, map3, map4, map5);
}
Also used : FreqMap(uk.gov.gchq.gaffer.types.FreqMap) PredicateMap(uk.gov.gchq.koryphe.predicate.PredicateMap) IsMoreThan(uk.gov.gchq.koryphe.impl.predicate.IsMoreThan)

Aggregations

FreqMap (uk.gov.gchq.gaffer.types.FreqMap)62 Test (org.junit.jupiter.api.Test)29 ArrayList (java.util.ArrayList)9 Edge (uk.gov.gchq.gaffer.data.element.Edge)9 Entity (uk.gov.gchq.gaffer.data.element.Entity)9 Date (java.util.Date)6 Schema (uk.gov.gchq.gaffer.store.schema.Schema)6 FunctionTest (uk.gov.gchq.koryphe.function.FunctionTest)6 Configuration (org.apache.hadoop.conf.Configuration)5 FileSystem (org.apache.hadoop.fs.FileSystem)5 Path (org.apache.hadoop.fs.Path)5 Row (org.apache.spark.sql.Row)5 SparkSession (org.apache.spark.sql.SparkSession)5 WrappedArray (scala.collection.mutable.WrappedArray)5 Element (uk.gov.gchq.gaffer.data.element.Element)5 HashMap (java.util.HashMap)4 Key (org.apache.accumulo.core.data.Key)4 Test (org.junit.Test)4 EdgeId (uk.gov.gchq.gaffer.data.element.id.EdgeId)4 ElementId (uk.gov.gchq.gaffer.data.element.id.ElementId)4