def testFilterSegmentsByCell(self):
    ssm = SegmentSparseMatrix(2048, 1000)

    # Shuffled [42, 42, 42, 43, 46, 47, 48]
    cellsWithSegments = [47, 42, 46, 43, 42, 48, 42]

    segments = ssm.createSegments(cellsWithSegments)
    ssm.sortSegmentsByCell(segments)

    # Include everything
    everything = sorted(cellsWithSegments)
    np.testing.assert_equal(ssm.filterSegmentsByCell(segments, everything),
                            segments)

    # Subset, one cell with multiple segments
    np.testing.assert_equal(ssm.filterSegmentsByCell(segments,
                                                     [42, 43, 48]),
                            [segments[0],
                             segments[1],
                             segments[2],
                             segments[3],
                             segments[6]])

    # Subset, some cells without segments
    np.testing.assert_equal(ssm.filterSegmentsByCell(segments,
                                                     [43, 44, 45, 48]),
                            [segments[3],
                             segments[6]])
Пример #2
0
  def testFilterSegmentsByCell(self):
    ssm = SegmentSparseMatrix(2048, 1000)

    # Shuffled [42, 42, 42, 43, 46, 47, 48]
    cellsWithSegments = [47, 42, 46, 43, 42, 48, 42]

    segments = ssm.createSegments(cellsWithSegments)
    ssm.sortSegmentsByCell(segments)

    # Include everything
    everything = sorted(cellsWithSegments)
    np.testing.assert_equal(ssm.filterSegmentsByCell(segments, everything),
                            segments)

    # Subset, one cell with multiple segments
    np.testing.assert_equal(ssm.filterSegmentsByCell(segments,
                                                     [42, 43, 48]),
                            [segments[0],
                             segments[1],
                             segments[2],
                             segments[3],
                             segments[6]])

    # Subset, some cells without segments
    np.testing.assert_equal(ssm.filterSegmentsByCell(segments,
                                                     [43, 44, 45, 48]),
                            [segments[3],
                             segments[6]])
    def testSortSegmentsByCell(self):
        ssm = SegmentSparseMatrix(2048, 1000)

        segments = ssm.createSegments([42, 41, 49, 45, 0, 2047])
        segmentsSorted = [
            segments[4], segments[1], segments[0], segments[3], segments[2],
            segments[5]
        ]

        ssm.sortSegmentsByCell(segments)

        np.testing.assert_equal(segments, segmentsSorted)
  def testSortSegmentsByCell(self):
    ssm = SegmentSparseMatrix(2048, 1000)

    segments = ssm.createSegments([42, 41, 49, 45, 0, 2047])
    segmentsSorted = [segments[4],
                      segments[1],
                      segments[0],
                      segments[3],
                      segments[2],
                      segments[5]]

    ssm.sortSegmentsByCell(segments)

    np.testing.assert_equal(segments, segmentsSorted)