import java.util.Random;
import java.util.Scanner;

public class FrequencyBagTester
{
	public static void main(String[] args)
	{
		int numberOfData = 20;
		int maxValueEx = 5;
		boolean wrongFrequency = false;
		int failed = 0;
		Scanner in = new Scanner(System.in);
		//CHANGE THIS BACK----------------------------------CHANGE THIS BACK-------
		int option = 1;
		//CHANGE THIS BACK----------------------------------CHANGE THIS BACK-------
		
		while(option == 0)
		{
			System.out.println("Please select a test option");
			System.out.println("  1. Stop the test as soon as an error is detected");
			System.out.println("  2. Do not stop the test when an error is detected");
			System.out.print("Enter an option (1 or 2): ");
			option = in.nextInt();
			
			if(option != 1 && option != 2)
			{
				System.out.println(option + " is not a valid choice.\n");
				option = 0;
			}
		}
		
		in.close();
		
		FrequencyBag<Integer> fb = new FrequencyBag<Integer>();
		
		// getFrequencyOf() empty bag
		
		System.out.print("Checking the method getFrequencyOf() of an empty frequency bag: ");
		for(int i = 0; i < maxValueEx; i++)
		{
			if(fb.getFrequencyOf(i) != 0)
			{
				if(!wrongFrequency)
				{
					System.out.println("FAIL");
					failed++;
				}
				System.out.println("The method getFrequencyOf(" + i + ") of an empty bag should return 0.");
				System.out.println("But your method getFrequencyOf(" + i + ") returns " + fb.getFrequencyOf(i) + ".");
				wrongFrequency = true;
			}
		}
		
		if(!wrongFrequency)
		{
			System.out.println("PASS");
		}
		else if(option == 1)
		{
			return;
		}
		
		
		// size() of empty bag.
		
		System.out.print("Checking the method size() of an empty bag: ");
		if(fb.size() != 0)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("The size of an empty bag should be 0.");
			System.out.println("But your method size() returns " + fb.size() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		// getMaxFrequency() of empty bag
		
		System.out.print("Checking the method getMaxFrequency() of an empty frequency bag: ");
		if(fb.getMaxFrequency() != 0)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("The method getMaxFrequency() of an empty bag should return 0.");
			System.out.println("But your method getMaxFrequency() returns " + fb.getMaxFrequency() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS\n");
		}
		
		// Adding data into empty bag
		
		int dataSet[] = new int[5];
		
		for(int i = 0; i < maxValueEx; i++)
		{
			dataSet[i] = 0;
		}
		
		Random rand = new Random();
		
		System.out.println("Add the following data into your frequency bag (in that order):");
		
		for(int i = 0; i < numberOfData; i++)
		{
			int temp = rand.nextInt(maxValueEx);
			
			System.out.print(temp + " ");
			fb.add(temp);
			dataSet[temp]++;
		}
		System.out.println();
		
		// getFrequencyOf()
		
		System.out.println("Checking the method getFrequencyOf() of each data in this frequency bag: ");
		
		for(int i = 0; i < maxValueEx; i++)
		{
			System.out.print("Frequency of " + i + ": ");
			if(fb.getFrequencyOf(i) != dataSet[i])
			{
				failed++;
				System.out.println("FAIL");
				System.out.println("The frequency of " + i + " is " + dataSet[i] + ".");
				System.out.println("But the method getFrequencyOf(" + i + ") returns " + fb.getFrequencyOf(i) + ".\n");
				if(option == 1)
				{
					return;
				}
			}
			else
			{
				System.out.println("PASS");
			}
		}

		// getFrequencyOf()
		
		System.out.print("Checking the method getFrequencyOf() of a data that is not in this frequency bag: ");
		
		if(fb.getFrequencyOf(-1) != 0)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("The frequency of -1 is 0.");
			System.out.println("But the method getFrequencyOf(-1) returns " + fb.getFrequencyOf(-1) + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}

		// getProbabilityOf()
		
		System.out.println("Checking the method getProbabilityOf() of each data in this frequency bag: ");
		
		for(int i = 0; i < maxValueEx; i++)
		{
			System.out.print("Probability of " + i + ": ");
			if(Math.abs(fb.getProbabilityOf(i) - ((double) dataSet[i]/numberOfData)) > 0.000001)
			{
				failed++;
				System.out.println("FAIL");
				System.out.println("The probability of " + i + " is " + ((double) dataSet[i]/numberOfData) + ".");
				System.out.println("But the method getProbabilityOf(" + i + ") returns " + fb.getProbabilityOf(i) + ".\n");
				if(option == 1)
				{
					return;
				}
			}
			else
			{
				System.out.println("PASS");
			}
		}
		
		// getProbabilityOf()
		
		System.out.print("Checking the method getProbabilityOf() of a data that is not in this frequency bag: ");
		
		if(fb.getProbabilityOf(-1) != 0.0)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("The probability of -1 is 0.");
			System.out.println("But the method getProbabilityOf(-1) returns " + fb.getProbabilityOf(-1) + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		// size()

		System.out.print("Checking the method size() of this non-empty frequency bag: ");
		if(fb.size() != numberOfData)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("After adding " + numberOfData + " entries, the method size() should return " + numberOfData + ".");
			System.out.println("But your method size() returns " + fb.size() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		// getMaxFrequency()
		
		System.out.print("Checking the method getMaxFrequency() of this non-empty frequency bag: ");
		
		int tempMax = 0;
		
		for(int i = 0; i < maxValueEx; i++)
		{
			if(tempMax < dataSet[i])
			{
				tempMax = dataSet[i];
			}
		}
		
		if(fb.getMaxFrequency() != tempMax)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("The method getMaxFrequency() of this non-empty frequency bag should return " + tempMax + ".");
			System.out.println("But your method getMaxFrequency() of this non-empty frequency bag returns " + fb.getMaxFrequency() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS\n");
		}
		
		// clear();
		
		fb.clear();
		
		System.out.println("Checking the method clear()");
		System.out.print("Checking the method size() after clear(): ");
		if(fb.size() != 0)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("After the bag is cleared, the method size() should return 0.");
			System.out.println("But your method size() returns " + fb.size() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		// getFrequencyOf() after clear()
		
		System.out.print("Checking the method getFrequencyOf(3) after clear(): ");
		if(fb.getFrequencyOf(3) != 0)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("After the bag is cleared, the method getFrequencyOf(3) should return 0.");
			System.out.println("But your method getFrequencyOf(3) returns " + fb.getFrequencyOf(3) + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		// getMaxFrequency() after clear()
		
		System.out.print("Checking the method getMaxFrequency() after clear(): ");
		if(fb.getMaxFrequency() != 0)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("After the bag is cleared, the method getMaxFrequency() should return 0.");
			System.out.println("But your method getMaxFrequency() returns " + fb.getMaxFrequency() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		// Generate large bag
		
		numberOfData = 1000000;
		maxValueEx = 1000;
		
		dataSet = new int[maxValueEx];
		
		for(int i = 0; i < maxValueEx; i++)
		{
			dataSet[i] = 0;
		}
		
		System.out.println("\nSo far so good.");
		System.out.println("Let's try adding " + numberOfData + " data into your frequency bag.");
		
		for(int i = 0; i < numberOfData; i++)
		{
			int temp = rand.nextInt(maxValueEx);
			
			fb.add(temp);
			dataSet[temp]++;
		}

		// size() of large bag.
		
		System.out.print("Checking the method size() of this non-empty frequency bag: ");
		if(fb.size() != numberOfData)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("After adding " + numberOfData + " entries, the method size() should return " + numberOfData + ".");
			System.out.println("But your method size() returns " + fb.size() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}

		// getFrequencyOf() of large bag
		
		System.out.print("Checking the method getFrequencyOf() of each data in this frequency bag: ");

		wrongFrequency = false;
		
		for(int i = 0; i < maxValueEx; i++)
		{
			if(fb.getFrequencyOf(i) != dataSet[i])
			{
				wrongFrequency = true;
			}
		}

		if(wrongFrequency)
		{
			failed++;
			System.out.println("FAIL");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		// getProbabilityOf() of large bag
		
		System.out.print("Checking the method getProbabilityOf() of each data in this frequency bag: ");

		boolean wrongProbability = false;
		
		for(int i = 0; i < maxValueEx; i++)
		{
			if(Math.abs(fb.getProbabilityOf(i) - ((double) dataSet[i]/numberOfData)) > 0.000001)
			{
				wrongProbability = true;
			}
		}

		if(wrongProbability)
		{
			failed++;
			System.out.println("FAIL");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		// getMaxFrequency() of large bag
		
		System.out.print("Checking the method getMaxFrequency() of this non-empty frequency bag: ");
		
		tempMax = 0;
		
		for(int i = 0; i < maxValueEx; i++)
		{
			if(tempMax < dataSet[i])
			{
				tempMax = dataSet[i];
			}
		}
		
		if(fb.getMaxFrequency() != tempMax)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("The method getMaxFrequency() of this non-empty frequency bag should return " + tempMax + ".");
			System.out.println("But your method getMaxFrequency() of this non-empty frequency bag returns " + fb.getMaxFrequency() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}

		// Frequency Bag of String
		
		System.out.println("\nSo far so good.");
		System.out.println("Let's construct a frequency bag of String.");
		
		FrequencyBag<String> sfb = new FrequencyBag<String>();
		
		String s = "hello how are you i am find thank you and you i am fine thank you";
		String[] str = s.split(" ");
		
		for(int i = 0; i < str.length; i++)
		{
			sfb.add(str[i]);
		}
		
		System.out.println("Adding the following strings into an empty frequency bag:");
		System.out.println(s + "\n");
		
		System.out.print("Checking the method size() of this non-empty frequency bag: ");
		if(sfb.size() != 16)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("After adding 16 entries, the method size() should return 16.");
			System.out.println("But your method size() returns " + sfb.size() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		System.out.print("Checking the method getFrequencyOf(\"am\"): ");
		if(sfb.getFrequencyOf("am") != 2)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("There are two strings \"am\" added into this bag.");
			System.out.println("But your method getFrequencyOf(\"am\") returns " + sfb.getFrequencyOf("am") + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		System.out.print("Checking the method getMaxFrequency(): ");
		if(sfb.getMaxFrequency() != 4)
		{
			failed++;
			System.out.println("FAIL");
			System.out.println("The maximum frequency should be 4 (the string \"you\").");
			System.out.println("But your method getMaxFrequency() returns " + sfb.getMaxFrequency() + ".\n");
			if(option == 1)
			{
				return;
			}
		}
		else
		{
			System.out.println("PASS");
		}
		
		System.out.println();
		
		if(failed != 0)
		{
			System.out.println("Threre are " + failed + " in this test. Fix your program.");
		}
		else
		{
			System.out.println("Congratulation!!! Your FrequencyBag works perfectly (I hope).");
			System.out.println("Run the program FrequencyFrame and compare its result.");
		}
	}
}