[FIX] comment assignment fails sometimes (#130)

Author: PabloRMira

README.md

@@ -48,7 +48,7 @@ repos:
 
 To exemplify the formatting let's say you have a SQL query like this
 
-```python
+```
 example_sql = """
 create or replace table mytable as -- mytable example
 seLecT a.asdf, b.qwer, -- some comment here
@@ -67,7 +67,7 @@ groUp by a.asdf
 
 Then you can use this package to format it so that it is better readable
 
-```python
+```
 from sql_formatter.core import format_sql
 print(format_sql(example_sql))
 ```
@@ -91,7 +91,7 @@ print(format_sql(example_sql))
 
 It can even deal with subqueries and it will correct my favourite simple careless mistake (comma at the end of SELECT statement before of FROM) for you on the flow :-)
 
-```python
+```
 print(format_sql("""
 select asdf, cast(qwer as numeric), -- some comment
 qwer1
@@ -121,7 +121,7 @@ where qwer1 >= 0
 
 The formatter is also robust against nested subqueries
 
-```python
+```
 print(format_sql("""
 select field1, field2 from (select field1, 
 field2 from (select field1, field2, 
@@ -143,7 +143,7 @@ field3 from table1 where a=1 and b>=100))
 
 If you do not want to get some query formatted in your SQL file then you can use the marker `/*skip-formatter*/` in your query to disable formatting for just the corresponding query
 
-```python
+```
 from sql_formatter.format_file import format_sql_commands
 print(format_sql_commands(
 """

docs/core.html

@@ -1750,6 +1750,100 @@ <h4 id="format_select" class="doc_header"><code>format_select</code><a href="htt
 </div>
 </div>
 
+</div>
+    {% endraw %}
+
+    {% raw %}
+    
+<div class="cell border-box-sizing code_cell rendered">
+<div class="input">
+
+<div class="inner_cell">
+    <div class="input_area">
+<div class=" highlight hl-ipython3"><pre><span></span><span class="n">assert_and_print</span><span class="p">(</span>
+    <span class="n">format_select</span><span class="p">(</span><span class="s2">&quot;select asdf, -- Some comment[C]asdforqwer -- Another comment[C]&quot;</span><span class="p">),</span>
+<span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">select asdf, -- Some comment</span>
+<span class="sd">       asdforqwer -- Another comment</span>
+<span class="sd">&quot;&quot;&quot;</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
+<span class="p">)</span>
+</pre></div>
+
+    </div>
+</div>
+</div>
+
+<div class="output_wrapper">
+<div class="output">
+
+<div class="output_area">
+
+<div class="output_subarea output_stream output_stdout output_text">
+<pre>Assertion failed
+
+Observed:
+
+select asdf, -- Some comment -- Another comment
+       asdforqwer
+
+
+Expected:
+
+select asdf, -- Some comment
+       asdforqwer -- Another comment
+Exception found at position 28:
+
+10-characters window:
+
+Observed:
+
+mment -- A
+
+
+Expected:
+
+mment
+    
+</pre>
+</div>
+</div>
+
+<div class="output_area">
+
+<div class="output_subarea output_text output_error">
+<pre>
+<span class="ansi-red-fg">-----------------------------------------------------------------------</span>
+<span class="ansi-red-fg">AssertionError</span>                        Traceback (most recent call last)
+<span class="ansi-green-fg">~/Dokumente/Python_Stuff/Projekte/sql_formatter/sql_formatter/utils.py</span> in <span class="ansi-cyan-fg">assert_and_print</span><span class="ansi-blue-fg">(s_in, s_expected)</span>
+<span class="ansi-green-intense-fg ansi-bold">     18</span>     <span class="ansi-green-fg">try</span><span class="ansi-blue-fg">:</span>
+<span class="ansi-green-fg">---&gt; 19</span><span class="ansi-red-fg">         </span><span class="ansi-green-fg">assert</span> s_in <span class="ansi-blue-fg">==</span> s_expected
+<span class="ansi-green-intense-fg ansi-bold">     20</span>     <span class="ansi-green-fg">except</span><span class="ansi-blue-fg">:</span>
+
+<span class="ansi-red-fg">AssertionError</span>: 
+
+During handling of the above exception, another exception occurred:
+
+<span class="ansi-red-fg">AssertionError</span>                        Traceback (most recent call last)
+<span class="ansi-green-fg">&lt;ipython-input-70-57dc18c8340b&gt;</span> in <span class="ansi-cyan-fg">&lt;module&gt;</span>
+<span class="ansi-green-intense-fg ansi-bold">      4</span> select asdf<span class="ansi-blue-fg">,</span> <span class="ansi-blue-fg">-</span><span class="ansi-blue-fg">-</span> Some comment
+<span class="ansi-green-intense-fg ansi-bold">      5</span>        asdforqwer <span class="ansi-blue-fg">-</span><span class="ansi-blue-fg">-</span> Another comment
+<span class="ansi-green-fg">----&gt; 6</span><span class="ansi-red-fg"> &#34;&#34;&#34;.strip()
+</span><span class="ansi-green-intense-fg ansi-bold">      7</span> )
+
+<span class="ansi-green-fg">~/Dokumente/Python_Stuff/Projekte/sql_formatter/sql_formatter/utils.py</span> in <span class="ansi-cyan-fg">assert_and_print</span><span class="ansi-blue-fg">(s_in, s_expected)</span>
+<span class="ansi-green-intense-fg ansi-bold">     35</span>         print<span class="ansi-blue-fg">(</span><span class="ansi-blue-fg">&#34;Expected:\n&#34;</span><span class="ansi-blue-fg">)</span>
+<span class="ansi-green-intense-fg ansi-bold">     36</span>         print<span class="ansi-blue-fg">(</span>s_expected<span class="ansi-blue-fg">[</span>max<span class="ansi-blue-fg">(</span>i<span class="ansi-blue-fg">-</span><span class="ansi-cyan-fg">5</span><span class="ansi-blue-fg">,</span> <span class="ansi-cyan-fg">0</span><span class="ansi-blue-fg">)</span><span class="ansi-blue-fg">:</span>i<span class="ansi-blue-fg">+</span><span class="ansi-cyan-fg">5</span><span class="ansi-blue-fg">]</span><span class="ansi-blue-fg">)</span>
+<span class="ansi-green-fg">---&gt; 37</span><span class="ansi-red-fg">         </span><span class="ansi-green-fg">assert</span> s_in <span class="ansi-blue-fg">==</span> s_expected
+<span class="ansi-green-intense-fg ansi-bold">     38</span>     print<span class="ansi-blue-fg">(</span>s_in<span class="ansi-blue-fg">)</span>
+<span class="ansi-green-intense-fg ansi-bold">     39</span>     <span class="ansi-green-fg">return</span> <span class="ansi-green-fg">None</span>
+
+<span class="ansi-red-fg">AssertionError</span>: </pre>
+</div>
+</div>
+
+</div>
+</div>
+
 </div>
     {% endraw %}