From 9f53a1eb87ebc87232e938b63ec263f63f267284 Mon Sep 17 00:00:00 2001
From: Bogdan Condorachi <bogdan.condorachi@gmail.com>
Date: Fri, 17 May 2024 01:05:27 +0300
Subject: [PATCH] Add python unit test

---
 tests/Languages/Python/PythonLanguageTest.php | 58 +++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 tests/Languages/Python/PythonLanguageTest.php

diff --git a/tests/Languages/Python/PythonLanguageTest.php b/tests/Languages/Python/PythonLanguageTest.php
new file mode 100644
index 0000000..f4d5e4e
--- /dev/null
+++ b/tests/Languages/Python/PythonLanguageTest.php
@@ -0,0 +1,58 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Tempest\Highlight\Tests\Languages\Python;
+
+use PHPUnit\Framework\Attributes\DataProvider;
+use PHPUnit\Framework\TestCase;
+use Tempest\Highlight\Highlighter;
+
+class PythonLanguageTest extends TestCase
+{
+    #[DataProvider('data')]
+    public function test_highlight(string $content, string $expected): void
+    {
+        $highlighter = new Highlighter();
+
+        $this->assertSame(
+            $expected,
+            $highlighter->parse($content, 'python'),
+        );
+
+        $this->assertSame(
+            $expected,
+            $highlighter->parse($content, 'py'),
+        );
+    }
+
+    public static function data(): array
+    {
+        return [
+            [<<<TXT
+def fib(n):    # write Fibonacci series up to n
+    """Print a Fibonacci series up to n."""
+    a, b = 0, 1
+    while a < n:
+        print(a, end=' ')
+        a, b = b, a+b
+    print()
+        
+# Now call the function we just defined:
+fib(2000)
+TXT,
+            <<<TXT
+<span class="hl-keyword">def</span> <span class="hl-property">fib</span>(n):    <span class="hl-comment"># write Fibonacci series up to n</span>
+    <span class="hl-value">&quot;&quot;&quot;Print a Fibonacci series up to n.&quot;&quot;&quot;</span>
+    a, b <span class="hl-operator">=</span> 0, 1
+    <span class="hl-keyword">while</span> a <span class="hl-operator">&lt;</span> n:
+        <span class="hl-type">print</span>(a, <span class="hl-variable">end</span><span class="hl-operator">=</span>'<span class="hl-value"> </span>')
+        a, b <span class="hl-operator">=</span> b, a<span class="hl-operator">+</span>b
+    <span class="hl-type">print</span>()
+
+<span class="hl-comment"># Now call the function we just defined:</span>
+fib(2000)
+TXT],
+        ];
+    }
+}