From e0c173378bb45e2aa3455ea044b914c7dcffc8b4 Mon Sep 17 00:00:00 2001 From: Oscar Najera Date: Fri, 1 Dec 2023 18:34:11 +0100 Subject: server cleanups --- scratch/semgrep/tests.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 scratch/semgrep/tests.py (limited to 'scratch/semgrep/tests.py') diff --git a/scratch/semgrep/tests.py b/scratch/semgrep/tests.py new file mode 100644 index 0000000..0e5d233 --- /dev/null +++ b/scratch/semgrep/tests.py @@ -0,0 +1,15 @@ +#!/usr/bin/env python3 +import server as s + + +def test_drop_dup(): + sample = [ + {"document": "Hello", "metadata": 5}, + {"document": "World", "metadata": 8}, + {"document": "Hello", "metadata": 6}, + {"document": "Good", "metadata": 3}, + {"document": "World", "metadata": 9}, + {"document": "World\n\n", "metadata": 9}, + ] + + assert s.drop_duplicates(sample) == (["Hello", "World", "Good"], [5, 8, 3]) -- cgit v1.2.3