Fix task numbers and add kata magic commands to the Jupyter notebook

Author: kurabirko

KeyDistribution_E91/KeyDistribution_E91.ipynb

@@ -31,6 +31,8 @@
    "cell_type": "code",
    "execution_count": null,
    "source": [
+    "%kata T11_EntangledPairst\n",
+    "\n",
     "operation EntangledPairs (qsAlice : Qubit[], qsBob : Qubit[]) : Unit is Adj{\n",
     "    Fact(Length(qsAlice) == Length(qsBob), \"Alice and Bob should have the same number of qubits\");\n",
     "    // ...\n",
@@ -107,6 +109,8 @@
    "cell_type": "code",
    "execution_count": null,
    "source": [
+    "%kata T21_RotateAndMeasure\n",
+    "\n",
     "operation RotateAndMeasure(q : Qubit, basisIndex: Int) : Result {\n",
     "    // ...\n",
     "\n",
@@ -147,6 +151,8 @@
    "cell_type": "code",
    "execution_count": null,
    "source": [
+    "%kata T22_RandomBasesArray\n",
+    "\n",
     "operation RandomBasesArray (basesIndices: Int[], N: Int) : Int[] {\n",
     "    // ...\n",
     "\n",
@@ -174,6 +180,8 @@
    "cell_type": "code",
    "execution_count": null,
    "source": [
+    "%kata T23_MeasureQubitArray\n",
+    "\n",
     "operation MeasureQubitArray (qs: Qubit[], basesIndices: Int[]) : Result[] {\n",
     "    // ...\n",
     "\n",
@@ -203,6 +211,8 @@
    "cell_type": "code",
    "execution_count": null,
    "source": [
+    "%kata T24_GenerateSharedKey\n",
+    "\n",
     "function GenerateSharedKey (basesAlice: Int[], basesBob: Int[], results: Result[]) : Bool[] {\n",
     "    let key = new Bool[0];\n",
     "    // ...\n",
@@ -226,7 +236,7 @@
    "cell_type": "code",
    "execution_count": null,
    "source": [
-    "operation T24_E91Protocol () : Unit {\n",
+    "operation T25_E91Protocol () : Unit {\n",
     "    // 1. Alice and Bob are distributed arrays of entangled pairs\n",
     "\n",
     "    // 2. Alice and Bob choose random measurement bases\n",
@@ -244,7 +254,7 @@
    "cell_type": "code",
    "execution_count": null,
    "source": [
-    "%simulate T24_E91Protocol"
+    "%simulate T25_E91Protocol"
    ],
    "outputs": [],
    "metadata": {}
@@ -283,6 +293,8 @@
    "cell_type": "code",
    "execution_count": null,
    "source": [
+    "%kata T31_CorrelationCheck\n",
+    "\n",
     "function CorrelationCheck(basesAlice: Int[], basesBob: Int[], resultsAlice: Result[], resultsBob: Result[]) : Double {\n",
     "    // ...\n",
     "\n",
@@ -323,6 +335,8 @@
    "cell_type": "code",
    "execution_count": null,
    "source": [
+    "%kata T32_Eavesdrop\n",
+    "\n",
     "operation Eavesdrop (qAlice : Qubit, qBob : Qubit, basisIndex : Int) : (Result, Result) {\n",
     "    Fact(basisIndex == 2 or basisIndex == 3, \"Eve should measure in one of Alice's and Bob's compatible basis\");\n",
     "    // ...\n",

KeyDistribution_E91/ReferenceImplementation.qs

@@ -164,7 +164,7 @@ namespace Quantum.Kata.KeyDistributionE91 {
     }
 
     // Task 3.3. Catch the eavesdropper
-    operation T32_E91ProtocolWithEavesdropper_Reference () : Unit {
+    operation T33_E91ProtocolWithEavesdropper_Reference () : Unit {
         // 1. Alice and Bob are distributed arrays of entangled pairs
         let N = 10;
         use (qsAlice, qsBob) = (Qubit[N] ,Qubit[N]);

KeyDistribution_E91/Tests.qs

@@ -31,6 +31,7 @@ namespace Quantum.Kata.KeyDistributionE91 {
 
         AssertAllZero(qsAlice + qsBob);
     }
+    
 
     //////////////////////////////////////////////////////////////////
     // Part II. E91 Protocol
@@ -50,7 +51,6 @@ namespace Quantum.Kata.KeyDistributionE91 {
             }
             Reset(q);
         }
-        
     }
 
     @Test("QuantumSimulator")
@@ -61,7 +61,7 @@ namespace Quantum.Kata.KeyDistributionE91 {
         let basesArray = RandomBasesArray(basesIndices, N);
         EqualityFactI(
             Length(basesArray), N,
-            "Generated array length does not match the input length"
+            $"Generated array should be of length {N}"
             );
 
         for i in basesArray {
@@ -125,11 +125,11 @@ namespace Quantum.Kata.KeyDistributionE91 {
         );
     }
 
+
     //////////////////////////////////////////////////////////////////
     // Part III. Eavesdropping
     //////////////////////////////////////////////////////////////////
 
-
     @Test("QuantumSimulator")
     function T31_CorrelationCheck() : Unit {
         // ...