File: test_655_dbscan.py

package info (click to toggle)
ezdxf 1.4.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 104,528 kB
  • sloc: python: 182,341; makefile: 116; lisp: 20; ansic: 4
file content (31 lines) | stat: -rw-r--r-- 757 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#  Copyright (c) 2022, Manfred Moitzi
#  License: MIT License

import pytest
from ezdxf.math.clustering import dbscan
from ezdxf.render import forms


def make_cluster(size, dx, dy, dz):
    return forms.cube().scale_uniform(size).translate(dx, dy, dz).vertices


def test_two_simple_cluster():
    c1 = make_cluster(1, 0, 0, 0)
    c2 = make_cluster(1.5, 5, 0, 0)
    points = list(c1)
    points.extend(c2)
    cluster = dbscan(points, radius=2)
    assert len(cluster) == 2
    cluster.sort()
    assert set(cluster[0]) == set(c1)
    assert set(cluster[1]) == set(c2)


def test_cluster_noise():
    cluster = dbscan(make_cluster(10, 0, 0, 0), radius=2)
    assert len(cluster) == 8  # all noise


if __name__ == "__main__":
    pytest.main([__file__])