コード例 #1
0
  @Test
  public void returnsConnectionBetweenTwoLayers() {
    Set<Layer> layers = new HashSet<Layer>();
    Layer topLayer = new Layer();
    Layer middleLayer = new Layer();

    Connections firstConnection = new FullyConnected(topLayer, middleLayer, 0, 0);

    topLayer.addConnection(firstConnection);

    layers.add(topLayer);
    layers.add(middleLayer);
    network.setLayers(layers);

    assertEquals(firstConnection, network.getConnection(topLayer, middleLayer));
    assertEquals(firstConnection, network.getConnection(middleLayer, topLayer));
  }
コード例 #2
0
  @Test
  public void returnsNullWhenThereIsntAConnectingLayer() {
    Set<Layer> layers = new HashSet<Layer>();
    Layer topLayer = new Layer();
    Layer middleLayer = new Layer();
    Layer bottomLayer = new Layer();

    Connections firstConnection = new FullyConnected(topLayer, middleLayer, 0, 0);

    topLayer.addConnection(firstConnection);

    layers.add(topLayer);
    layers.add(middleLayer);
    network.setLayers(layers);

    assertNull(network.getConnection(topLayer, bottomLayer));
    assertNull(network.getConnection(bottomLayer, topLayer));
  }